genesis-flow 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

{genesis_flow-1.0.0.dist-info → genesis_flow-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genesis-flow
-Version: 1.0.0
+Version: 1.0.1
 Summary: Genesis-Flow: Lightweight, secure MLflow fork for Genesis platform
 Maintainer-email: Databricks <mlflow-oss-maintainers@googlegroups.com>
 License: Copyright 2018 Databricks, Inc.  All rights reserved.
@@ -761,6 +761,32 @@ pre-commit install
 
 # Run tests
 pytest tests/
+
+# 1. Install build tools
+  pip install build twine
+
+  # 2. Build the package
+  python -m build
+
+  # 3. Upload to PyPI
+  python -m twine upload dist/*
+
+  # Or upload to TestPyPI first
+  python -m twine upload --repository testpypi dist/*
+
+  Before uploading, make sure you have:
+  - A PyPI account and API token
+  - Configure your credentials in ~/.pypirc:
+
+  [pypi]
+  username = __token__
+  password = pypi-<your-token>
+
+  [testpypi]
+  repository = https://test.pypi.org/legacy/
+  username = __token__
+  password = pypi-<your-token>
+
 ```
 
 ### Code Quality

{genesis_flow-1.0.0.dist-info → genesis_flow-1.0.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-genesis_flow-1.0.0.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
+genesis_flow-1.0.1.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
 mlflow/__init__.py,sha256=-_r__N5Afed81pLVtr2wKbHQIA0aj9u9n_7kWGxLWi4,11194
 mlflow/__main__.py,sha256=_PcdoxKehR_a2MI6GqBfzYzRCXZhVyDCSdbxDWVlWd4,39
 mlflow/cli.py,sha256=f1ObrWZ03HgRiRoVEE1Gffe-dGcSY7CxJyEFgb5VUMM,26137
@@ -610,7 +610,7 @@ mlflow/utils/requirements_utils.py,sha256=79BL9iFqg9H4VRYiqk-gTLlrfLyUY7KVG4BDV8
 mlflow/utils/rest_utils.py,sha256=yLeFOP135qV2OFcvtb_fFtj18MATRoxV4wJ5UshnEyk,26540
 mlflow/utils/search_logged_model_utils.py,sha256=JWczQxogo4u9Gr3gfoSVf7tVny1e-rxfAchh3Vqy6RU,4289
 mlflow/utils/search_utils.py,sha256=PgXtwlViHvmahY63otykh5iJVW0PR9DIwetrvCCBw2U,86142
-mlflow/utils/secure_loading.py,sha256=WCh8NC_gwrqEjkXE1_YTGY525pOHRIOYn0Dq1luhB8A,7983
+mlflow/utils/secure_loading.py,sha256=yJpKwrhzINmo0yVdp6fXvbNkZu_HEdwyB8deljGDXxk,10095
 mlflow/utils/security_validation.py,sha256=cLBrLNOoVJuAffO677v0m_v4i-01v7aLk-sJk87mpkQ,12830
 mlflow/utils/server_cli_utils.py,sha256=gbT5CVkOLorSw-y8bnNSBEP9mClcVTagtTN5SK5Azhw,2381
 mlflow/utils/spark_utils.py,sha256=zUbQIRAtwyU3rDJK8m7v42KO2TSGn28Coe_UYBEhIWA,395
@@ -638,8 +638,8 @@ mlflow/utils/autologging_utils/metrics_queue.py,sha256=bwpMX7Go6xFxrpYROi6rDBdVt
 mlflow/utils/autologging_utils/safety.py,sha256=IwbTbusyE87Hc4qkhhvMikoaZqX9kpr972FWS2B8goc,51465
 mlflow/utils/autologging_utils/versioning.py,sha256=2hSN4KXFWEJCcopDdLG6BiPeqSoqjETeNMuUBsCCwlI,3762
 mlflow/utils/import_hooks/__init__.py,sha256=werje98Woelkbwrhtlb8wmRdt3RtiL--LqGru7Xh3YU,13589
-genesis_flow-1.0.0.dist-info/METADATA,sha256=D9XGiux4DI1WK3PrJNo2y4jdco5QfBUgVx8tGaQJ8rc,32051
-genesis_flow-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-genesis_flow-1.0.0.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
-genesis_flow-1.0.0.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
-genesis_flow-1.0.0.dist-info/RECORD,,
+genesis_flow-1.0.1.dist-info/METADATA,sha256=QOs7W_wCIn4bYsjQVwLX8JE9zoW1dBO4rEJsf_qQmzU,32586
+genesis_flow-1.0.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+genesis_flow-1.0.1.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
+genesis_flow-1.0.1.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
+genesis_flow-1.0.1.dist-info/RECORD,,

mlflow/utils/secure_loading.py

@@ -22,14 +22,14 @@ SAFE_PICKLE_CLASSES = {
     'numpy.dtype',
     'numpy.int32', 'numpy.int64', 'numpy.float32', 'numpy.float64',
     'numpy.bool_', 'numpy.str_',
-    
+
     # Pandas types
     'pandas.core.frame.DataFrame',
     'pandas.core.series.Series',
     'pandas.core.index.Index',
     'pandas.core.dtypes.dtypes.CategoricalDtype',
-    
-    # Scikit-learn estimators (core models only)
+
+    # Scikit-learn estimators
     'sklearn.linear_model._base.LinearRegression',
     'sklearn.linear_model._logistic.LogisticRegression',
     'sklearn.ensemble._forest.RandomForestClassifier',
@@ -38,20 +38,74 @@ SAFE_PICKLE_CLASSES = {
     'sklearn.tree._classes.DecisionTreeRegressor',
     'sklearn.svm._classes.SVC',
     'sklearn.svm._classes.SVR',
-    
-    # Standard Python types
+
+    # Built-in types
     'builtins.dict', 'builtins.list', 'builtins.tuple', 'builtins.set',
     'builtins.str', 'builtins.int', 'builtins.float', 'builtins.bool',
-    
+    'builtins.type',
+
     # Collections
     'collections.OrderedDict',
     'collections.defaultdict',
-    
-    # Genesis-Flow internal types
+
+    # MLflow types
     'mlflow.models.signature.ModelSignature',
+    'mlflow.models.signature._TypeHints',
     'mlflow.types.schema.Schema',
+    'mlflow.pyfunc.model.PythonModel',
+
+    # Cloudpickle internals
+    'cloudpickle.cloudpickle._make_skeleton_class',
+    'cloudpickle.cloudpickle._class_setstate',
+    'cloudpickle.cloudpickle._make_function',
+    'cloudpickle.cloudpickle._builtin_type',
+    'cloudpickle.cloudpickle._function_setstate',
+    'cloudpickle.cloudpickle._make_empty_cell',
+    'cloudpickle.cloudpickle._make_cell',
+
+    # Sentence Transformers
+    'sentence_transformers.SentenceTransformer.SentenceTransformer',
+    'sentence_transformers.model_card.SentenceTransformerModelCardData',
+    'sentence_transformers.models.Transformer.Transformer',
+    'sentence_transformers.models.Pooling.Pooling',
+    'sentence_transformers.models.Normalize.Normalize',
+
+    # Torch
+    'torch.torch_version.TorchVersion',
+    'torch._utils._rebuild_tensor_v2',
+    'torch.storage._load_from_bytes',
+    'torch.nn.modules.sparse.Embedding',
+    'torch._utils._rebuild_parameter',
+    'torch.nn.modules.normalization.LayerNorm',
+    'torch.nn.modules.dropout.Dropout',
+    'torch.nn.modules.container.ModuleList',
+    'torch.nn.modules.linear.Linear',
+    'torch.nn.modules.activation.Tanh',
+    'torch.float32',
+    'torch._C._nn.gelu',
+
+    # Transformers
+    'transformers.models.bert.modeling_bert.BertModel',
+    'transformers.models.bert.modeling_bert.BertEmbeddings',
+    'transformers.models.bert.modeling_bert.BertEncoder',
+    'transformers.models.bert.modeling_bert.BertLayer',
+    'transformers.models.bert.modeling_bert.BertAttention',
+    'transformers.models.bert.modeling_bert.BertSdpaSelfAttention',
+    'transformers.models.bert.modeling_bert.BertSelfOutput',
+    'transformers.models.bert.modeling_bert.BertIntermediate',
+    'transformers.models.bert.modeling_bert.BertOutput',
+    'transformers.models.bert.modeling_bert.BertPooler',
+    'transformers.models.bert.configuration_bert.BertConfig',
+    'transformers.models.bert.tokenization_bert_fast.BertTokenizerFast',
+    'transformers.activations.GELUActivation',
+
+    # Tokenizers
+    'tokenizers.Tokenizer',
+    'tokenizers.models.Model',
+    'tokenizers.AddedToken',
 }
 
+
 class RestrictedUnpickler(pickle.Unpickler):
     """
     Secure unpickler that only allows safe, whitelisted classes.