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

code_loader/inner_leap_binder/leapbinder.py

@@ -690,9 +690,10 @@ class LeapBinder:
             preprocess_response.tl_generated = True
             if not preprocess_response.length or preprocess_response.length < 1:
                 raise Exception("Simulation '{}' returned PreprocessResponse with length < 1".format(sim.name))
+            preprocess_response.sample_ids = [0]
             for handler in self.setup_container.inputs:
-                out1 = handler.function(0, preprocess_response)
-                out2 = handler.function(0, preprocess_response)
+                out1 = handler.function(preprocess_response.sample_ids[0], preprocess_response)
+                out2 = handler.function(preprocess_response.sample_ids[0], preprocess_response)
                 if not np.array_equal(out1, out2):
                     raise Exception(
                         "Simulation '{}': encoder '{}' is non-deterministic — consecutive calls with seed=0 returned different outputs".format(

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -2175,7 +2175,7 @@ def tensorleap_status_table():
         row = _find_row("tensorleap simulation")
         if not row:
             return
-        if row["Added to integration"] == CROSS:
+        if _crashed["value"]:
             return
         failed = [n for n, s in _sim_tracking.items() if s == "failed"]
         not_called = [n for n, s in _sim_tracking.items() if s == "registered"]

code_loader/leaploader.py

@@ -324,9 +324,10 @@ class LeapLoader(LeapLoaderBase):
                 preprocess_response.tl_generated = True
                 if preprocess_response.length < 1:
                     raise ValueError("Simulation returned PreprocessResponse with length < 1")
+                preprocess_response.sample_ids = [0]
                 for handler in global_leap_binder.setup_container.inputs:
-                    out1 = handler.function(0, preprocess_response)
-                    out2 = handler.function(0, preprocess_response)
+                    out1 = handler.function(preprocess_response.sample_ids[0], preprocess_response)
+                    out2 = handler.function(preprocess_response.sample_ids[0], preprocess_response)
                     if not np.array_equal(out1, out2):
                         raise ValueError(
                             "Encoder '{}' is non-deterministic: consecutive calls with seed=0 returned different outputs".format(
@@ -363,9 +364,10 @@ class LeapLoader(LeapLoaderBase):
             _simulation_context["active"] = False
         sim_preprocess.state = DataStateType.additional
         sim_preprocess.tl_generated = True
+        sim_preprocess.sample_ids = [0]
         result = {}
         for handler in global_leap_binder.setup_container.inputs:
-            result[handler.name] = handler.function(0, sim_preprocess)
+            result[handler.name] = handler.function(sim_preprocess.sample_ids[0], sim_preprocess)
         return result
 
     @staticmethod

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

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

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

@@ -21,9 +21,9 @@ code_loader/experiment_api/types.py,sha256=MY8xFARHwdVA7p4dxyhD60ShmttgTvb4qdp1o
 code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZJEqBqc,3304
 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=xu0K72zS4b3Nkc7OqcYV0CJ-LNgkZ979RyDL0EVSuqM,40381
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=94D0fcUlrYV9h-Q4E16d8Cac6uFleNV3If3AKlKOWuE,105800
-code_loader/leaploader.py,sha256=-uN0T8x563uTJfq51f3osory8aON6MMBf1tpdT1Ox5I,36314
+code_loader/inner_leap_binder/leapbinder.py,sha256=cbw1zSAYeaRFTvcPPs5cNszs3LK8I-AiP9oJsK5A-pc,40494
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=YlK51b5Ryo396f7tOA7Ole3vYCLs3f5ZLm2qDQ9K1NE,105781
+code_loader/leaploader.py,sha256=V1GXbUwY1a8Cxo6AY1h2BLutzFPTLE7iLxX4wyMsVJQ,36498
 code_loader/leaploaderbase.py,sha256=0JMwslRs6w7psIkj4pTjYHd_pSio3vt5H-X0Lar2nBI,6428
 code_loader/mixpanel_tracker.py,sha256=rNwRmFifNbdUoqLQvvhhgpKczWpWiEmd8MfyJe27sxw,9131
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -32,7 +32,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=YecipkdTA-VcE9F0RQcY9cFnY8P3AksPnHM2Db7xUSk,3972
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.181.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.181.dev3.dist-info/METADATA,sha256=c-fjPMP1vvj24a7thpF3vs_248IScUU31Rk073QU7xQ,1095
-code_loader-1.0.181.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.181.dev3.dist-info/RECORD,,
+code_loader-1.0.181.dev5.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.181.dev5.dist-info/METADATA,sha256=PS2dji6b9IWHKX0Ro16X_-4pX6zUNo00BDQ3GHF9C3k,1095
+code_loader-1.0.181.dev5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.181.dev5.dist-info/RECORD,,