genesis-flow 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

{genesis_flow-1.0.6.dist-info → genesis_flow-1.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genesis-flow
-Version: 1.0.6
+Version: 1.0.7
 Summary: Genesis-Flow: MLflow v3.1.4 compatible fork for Genesis platform
 Maintainer-email: Databricks <mlflow-oss-maintainers@googlegroups.com>
 License: Copyright 2018 Databricks, Inc.  All rights reserved.

{genesis_flow-1.0.6.dist-info → genesis_flow-1.0.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-genesis_flow-1.0.6.dist-info/licenses/LICENSE.txt,sha256=Y5U1Xebzka__NZlqMPtBsYm0mRpMtUmTrONatpoL-ig,11382
+genesis_flow-1.0.7.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
@@ -613,7 +613,7 @@ mlflow/utils/requirements_utils.py,sha256=79BL9iFqg9H4VRYiqk-gTLlrfLyUY7KVG4BDV8
 mlflow/utils/rest_utils.py,sha256=R5MZk0Ck358VeW-DKqpfSyDHbWjg_slQTpRfr4ZLDLM,26664
 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=yJpKwrhzINmo0yVdp6fXvbNkZu_HEdwyB8deljGDXxk,10095
+mlflow/utils/secure_loading.py,sha256=1d-Hf0QkU6WkGjLKvAKcXv7sLa6xl2utig1lT-Hhyp4,10262
 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
@@ -641,8 +641,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.6.dist-info/METADATA,sha256=_ofi_wrmOHLBu-0gjHwRW97dMrrVpnksdsLhUr1WrlU,33401
-genesis_flow-1.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-genesis_flow-1.0.6.dist-info/entry_points.txt,sha256=YZZiTHSYQpv8vou6gdqWI17piK9Pm0P3BmZ-xepUtnw,449
-genesis_flow-1.0.6.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
-genesis_flow-1.0.6.dist-info/RECORD,,
+genesis_flow-1.0.7.dist-info/METADATA,sha256=Z4t8Zn9OIxkH7d-hTozfYP0c8_-VB-WQ0KWo8XxyDHw,33401
+genesis_flow-1.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+genesis_flow-1.0.7.dist-info/entry_points.txt,sha256=YZZiTHSYQpv8vou6gdqWI17piK9Pm0P3BmZ-xepUtnw,449
+genesis_flow-1.0.7.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
+genesis_flow-1.0.7.dist-info/RECORD,,

mlflow/utils/secure_loading.py

@@ -17,6 +17,10 @@ logger = logging.getLogger(__name__)
 
 # Allowlist of safe classes that can be unpickled
 SAFE_PICKLE_CLASSES = {
+
+    # STD types
+    'time.time',
+
     # NumPy types
     'numpy.ndarray',
     'numpy.dtype',
@@ -62,6 +66,7 @@ SAFE_PICKLE_CLASSES = {
     'cloudpickle.cloudpickle._function_setstate',
     'cloudpickle.cloudpickle._make_empty_cell',
     'cloudpickle.cloudpickle._make_cell',
+    'cloudpickle.cloudpickle.subimport',
 
     # Sentence Transformers
     'sentence_transformers.SentenceTransformer.SentenceTransformer',
@@ -103,6 +108,11 @@ SAFE_PICKLE_CLASSES = {
     'tokenizers.Tokenizer',
     'tokenizers.models.Model',
     'tokenizers.AddedToken',
+
+    # PyTorch
+    'model.load_checkpoint',
+    'torch.device',
+    'model.SepClassifier',
 }