code-loader 1.0.139.dev3__py3-none-any.whl → 1.0.139.dev5__py3-none-any.whl

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -229,6 +229,10 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
             import keras
         except ImportError:
             keras = None
+        try:
+            import tensorflow as tf
+        except ImportError:
+            tf = None
         try:
             import onnxruntime
         except ImportError:
@@ -237,7 +241,10 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
         if not keras and not onnxruntime:
             raise AssertionError(err_message)
 
-        is_keras_model = bool(keras and isinstance(result, getattr(keras, "Model", tuple())))
+        is_keras_model = (
+                bool(keras and isinstance(result, getattr(keras, "Model", tuple())))
+                or bool(tf and isinstance(result, getattr(tf.keras, "Model", tuple())))
+        )
         is_onnx_model = bool(onnxruntime and isinstance(result, onnxruntime.InferenceSession))
 
         if not any([is_keras_model, is_onnx_model]):
@@ -1413,18 +1420,19 @@ def tensorleap_status_table():
     import atexit
     import sys
     import traceback
-
+    CHECK = "✅"
+    CROSS = "❌"
 
     table = [
-        {"name": "tensorleap_preprocess", "status": "x"},
-        {"name": "tensorleap_integration_test", "status": "x"},
-        {"name": "tensorleap_input_encoder", "status": "x"},
-        {"name": "tensorleap_gt_encoder", "status": "x"},
-        {"name": "tensorleap_load_model", "status": "x"},
-        {"name": "tensorleap_custom_loss", "status": "x"},
-        {"name": "tensorleap_custom_metric (optional)", "status": "x"},
-        {"name": "tensorleap_metadata (optional)", "status": "x"},
-        {"name": "tensorleap_custom_visualizer (optional)", "status": "x"},
+        {"name": "tensorleap_preprocess", "Added to integration": CROSS},
+        {"name": "tensorleap_integration_test", "Added to integration": CROSS},
+        {"name": "tensorleap_input_encoder", "Added to integration": CROSS},
+        {"name": "tensorleap_gt_encoder", "Added to integration": CROSS},
+        {"name": "tensorleap_load_model", "Added to integration": CROSS},
+        {"name": "tensorleap_custom_loss", "Added to integration": CROSS},
+        {"name": "tensorleap_custom_metric (optional)", "Added to integration": CROSS},
+        {"name": "tensorleap_metadata (optional)", "Added to integration": CROSS},
+        {"name": "tensorleap_custom_visualizer (optional)", "Added to integration": CROSS},
 
     ]
 
@@ -1432,19 +1440,19 @@ def tensorleap_status_table():
 
     def _print_table():
         ready_mess = "\nAll parts have been successfully set. If no errors accured, you can now push the project to the Tensorleap system."
-        not_ready_mess = "\nSome mandatory components have not been set yet. Recommended next implementation step: "
-        mandatory_ready_mess = "\nAll mandatory parts have been successfully set. If no errors accured, you can now push the project to the Tensorleap system or continue to the next optional implementation step: "
+        not_ready_mess = "\nSome mandatory components have not yet been added to the Integration test. Recommended next interface to add is: "
+        mandatory_ready_mess = "\nAll mandatory parts have been successfully set. If no errors accured, you can now push the project to the Tensorleap system or continue to the next optional reccomeded interface,adding: "
 
         name_width = max(len(row["name"]) for row in table)
-        status_width = max(len(row["status"]) for row in table)
-        header = f"{'Function Name'.ljust(name_width)} | {'Status'.ljust(status_width)}"
+        status_width = max(len(row["Added to integration"]) for row in table)
+        header = f"{'Decorator Name'.ljust(name_width)} | {'Added to integration'.ljust(status_width)}"
         sep = "-" * len(header)
         print("\n" + header)
         print(sep)
         ready=True
         for row in table:
-            print(f"{row['name'].ljust(name_width)} | {row['status'].ljust(status_width)}")
-            if row['status']=='x' and ready:
+            print(f"{row['name'].ljust(name_width)} | {row['Added to integration'].ljust(status_width)}")
+            if row['Added to integration']==CROSS and ready:
                 ready=False
                 next_step=row['name']
 
@@ -1453,7 +1461,7 @@ def tensorleap_status_table():
     def update_env_params(name: str, status: str = "✓"):
         for row in table:
             if row["name"].removesuffix(" (optional)") == name:
-                row["status"] = status
+                row["Added to integration"] = CHECK if status=="v" else CROSS
                 break
     def run_on_exit():
         if _finalizer_called["done"]:

{code_loader-1.0.139.dev3.dist-info → code_loader-1.0.139.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.139.dev3
+Version: 1.0.139.dev5
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.139.dev3.dist-info → code_loader-1.0.139.dev5.dist-info}/RECORD

@@ -21,7 +21,7 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=Q3D9yVM-GNEJfYRFvMV__BoZbcWOgnWKhrZXAv6Tu7o,33232
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=Rgt_9Kllc0yZVXPsk1yC1p2Yb2OMx3QXMQlQvuixpZ0,70055
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=I6V-Q9qRqhmBJSr_TvrGTcs2RZgi8y9nbLWtL93UH4I,70581
 code_loader/leaploader.py,sha256=6D6xZzMI6qSNIb3tuKLB3BbK5H8QS1_r7iQjIXO3OkM,29795
 code_loader/leaploaderbase.py,sha256=LIFcC6xo6V_iiGN3BjibXETu_l84EWM_WIOKAvkfTiM,4458
 code_loader/mixpanel_tracker.py,sha256=l9z_szKKQ7apEbdNZpGH1TKAiT_TsBHb9AQnePaWTyo,4942
@@ -31,7 +31,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=gXENTYpjdidq2dx0gVbXlErPeHoNs-4TYAZbLRe0y2c,2712
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.139.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.139.dev3.dist-info/METADATA,sha256=MQ-BRAy9_HLMKQnO0Dv0dHoZ8NGwbfd90eZULo_hcx4,1095
-code_loader-1.0.139.dev3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-code_loader-1.0.139.dev3.dist-info/RECORD,,
+code_loader-1.0.139.dev5.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.139.dev5.dist-info/METADATA,sha256=VuBCYX1zRt_iDqU9pMv0cixcnB8b39BvSRkNXOW6N_8,1095
+code_loader-1.0.139.dev5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+code_loader-1.0.139.dev5.dist-info/RECORD,,