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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: genesis-flow
-Version: 1.0.5
+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.5.dist-info → genesis_flow-1.0.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-genesis_flow-1.0.5.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
@@ -26,7 +26,7 @@ mlflow/azure/client.py,sha256=dUaMFj04OLrlyvYSRTHXF4CaLHCGUdSzhjM5u8ucv8g,11509
 mlflow/azure/config.py,sha256=0DwiEYN3TefmdaGO9KogNNflX_-0wOeHwKt2hMxKEdU,4491
 mlflow/azure/connection_factory.py,sha256=1hcwPmc5xpP_MS1t3CIlzYfJ_mzEn2GBxnz5TJ1sHOY,12391
 mlflow/azure/exceptions.py,sha256=3jQo_kimK4g94LmSCtzfRQqEOl6bDBtwAQ6utXbgw8k,494
-mlflow/azure/stores.py,sha256=MnZaPU2eOAerzGfcwSoyuMCkd9ad1cOmsb6fclgxF4Q,12890
+mlflow/azure/stores.py,sha256=EwcAnb7RolKU-cI11Ak1Rdsv6a-TOQJbAOOaaoIcKuE,12876
 mlflow/azure/utils.py,sha256=uU4OWU9ffz3KhLf0IgGqabZTTiAskLrNGVH5bIcwKUU,5563
 mlflow/bedrock/__init__.py,sha256=LNOczTpqmgbdek-anWMNisPGDcnn-VbWQm5VVxQ7FdA,1846
 mlflow/bedrock/_autolog.py,sha256=7c2mj9pK1aj-IKCySl74g6gGxze2Bldo8bHSIZAQj5M,7800
@@ -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.5.dist-info/METADATA,sha256=c2PrO48hnz6iVieC5srClmgBx4MN_61P3W_NvJvFsQE,33401
-genesis_flow-1.0.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-genesis_flow-1.0.5.dist-info/entry_points.txt,sha256=fwlE6mYWea9uCNIsmXdT5qMIOZ8rYbTvO6XcZR8fBbE,495
-genesis_flow-1.0.5.dist-info/top_level.txt,sha256=wm8UqYyUHI21EvrTDHb3eYICy0dOVDLBhAL-jp5zbuI,7
-genesis_flow-1.0.5.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,,

{genesis_flow-1.0.5.dist-info → genesis_flow-1.0.7.dist-info}/entry_points.txt

@@ -16,4 +16,3 @@ openai = mlflow.deployments.openai
 
 [mlflow.tracking_store]
 azure-postgres = mlflow.azure.stores:create_store
-postgresql = mlflow.azure.stores:create_store

mlflow/azure/stores.py

@@ -172,22 +172,21 @@ def create_store(store_uri: str, artifact_uri: Optional[str] = None):
         "auth_method=managed_identity" in store_uri
     )
     
-    # If auth is not enabled via config and not explicitly requested in URI, skip Azure auth
+    # Check if we need to enable auth based on explicit request in URI
+    if explicit_azure_request and not config.auth_enabled:
+        # URI explicitly requests Azure auth, so enable it
+        config.auth_enabled = True
+        if not config.auth_method or config.auth_method == AuthMethod.SQL_AUTH:
+            config.auth_method = AuthMethod.MANAGED_IDENTITY
+        logger.debug("Enabled Azure auth due to explicit request in URI")
+    
+    # Log the decision for debugging
     if not config.auth_enabled and not explicit_azure_request:
-        # Log for debugging
         logger.debug(
             "Azure auth not enabled: config.auth_enabled=%s, explicit_request=%s",
             config.auth_enabled,
             explicit_azure_request
         )
-        return None
-    
-    # If we reach here, Azure auth is either enabled or explicitly requested
-    # Make sure config reflects this
-    if explicit_azure_request and not config.auth_enabled:
-        config.auth_enabled = True
-        if not config.auth_method or config.auth_method == AuthMethod.SQL_AUTH:
-            config.auth_method = AuthMethod.MANAGED_IDENTITY
     
     if config.should_use_azure_auth:
         logger.info("Creating Azure-enabled tracking store via plugin: auth_method=%s", config.auth_method.value)

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',
 }