code-loader 1.0.180.dev5__tar.gz → 1.0.181__tar.gz

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/PKG-INFO

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

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/code_loader/contract/datasetclasses.py

@@ -13,7 +13,7 @@ from code_loader.contract.sim_config import SimConfig
 
 custom_latent_space_attribute = "custom_latent_space"
 
-_simulation_context: dict = {"active": False}
+_simulation_context: Dict[str, bool] = {"active": False}
 
 
 @dataclass

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/code_loader/contract/sim_config.py

@@ -66,6 +66,19 @@ def _parse_bounds(name: str, metadata_type: DatasetMetadataType, bounds_raw: Dic
     raise ValueError(f"Parameter '{name}' has unsupported metadata type: {metadata_type}.")
 
 
+def sample_sim_params(sim_config: SimConfig) -> Dict[str, Any]:
+    params: Dict[str, Any] = {}
+    for name, param_config in sim_config.items():
+        bounds = param_config.bounds
+        if isinstance(bounds, FloatBounds):
+            params[name] = (bounds.min + bounds.max) / 2.0
+        elif isinstance(bounds, IntBounds):
+            params[name] = int((bounds.min + bounds.max) / 2)
+        else:  # CategoricalBounds
+            params[name] = bounds.values[0]
+    return params
+
+
 def parse_sim_config(raw: Dict[str, Any]) -> SimConfig:
     if not raw:
         raise ValueError("sim_config must have at least one parameter.")

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/code_loader/inner_leap_binder/leapbinder.py

@@ -13,7 +13,7 @@ from code_loader.contract.datasetclasses import SectionCallableInterface, InputH
     CustomMultipleReturnCallableInterfaceMultiArgs, DatasetBaseHandler, custom_latent_space_attribute, \
     RawInputsForHeatmap, VisualizerHandlerData, MetricHandlerData, CustomLossHandlerData, SamplePreprocessResponse, \
     ElementInstanceMasksHandler, InstanceCallableInterface, CustomLatentSpaceHandler, InstanceMetricHandler, \
-    SimulationHandler
+    SimulationHandler, _simulation_context
 from code_loader.contract.enums import LeapDataType, DataStateEnum, DataStateType, MetricDirection, DatasetMetadataType
 from code_loader.contract.mapping import NodeConnection, NodeMapping, NodeMappingType
 from code_loader.contract.responsedataclasses import DatasetTestResultPayload, LeapAnalysisConfiguration
@@ -21,7 +21,7 @@ from code_loader.contract.visualizer_classes import map_leap_data_type_to_visual
 from code_loader.default_losses import loss_name_to_function
 from code_loader.default_metrics import metrics_names_to_functions_and_direction
 from code_loader.utils import to_numpy_return_wrapper, get_shape, to_numpy_return_masks_wrapper
-from code_loader.contract.sim_config import parse_sim_config
+from code_loader.contract.sim_config import parse_sim_config, sample_sim_params
 from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
     default_graph_visualizer, \
     default_image_visualizer, default_horizontal_bar_visualizer, default_word_visualizer, \
@@ -666,10 +666,46 @@ class LeapBinder:
                     continue
                 test_result = self.check_handler(preprocess_response, test_result, dataset_base_handler, state)
 
+    def check_simulations(self) -> None:
+        from code_loader.inner_leap_binder.leapbinder_decorators import _call_from_tl_platform
+        if not _call_from_tl_platform:
+            from code_loader.inner_leap_binder.leapbinder_decorators import set_current
+            set_current("tensorleap simulation")
+        for sim in self.setup_container.simulations:
+            kwargs = sample_sim_params(sim.sim_config)
+            kwargs["N"] = 1
+            kwargs["seed"] = 0
+            _simulation_context["active"] = True
+            try:
+                preprocess_response = sim.function(**kwargs)
+            finally:
+                _simulation_context["active"] = False
+            if not isinstance(preprocess_response, PreprocessResponse):
+                raise Exception(
+                    "Simulation '{}' returned {} instead of PreprocessResponse".format(
+                        sim.name, type(preprocess_response).__name__
+                    )
+                )
+            preprocess_response.state = DataStateType.additional
+            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(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(
+                            sim.name, handler.name
+                        )
+                    )
+
     def check(self) -> None:
         preprocess_result = self.get_preprocess_result()
         self.check_preprocess(preprocess_result)
         self.check_handlers(preprocess_result)
+        self.check_simulations()
         self.validate_ignore_latent_spaces()
         print("Successful!")
 

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -1403,22 +1403,22 @@ def tensorleap_simulation(name: str, sim_params: dict):
     def decorating_function(user_function: Callable[..., PreprocessResponse]):
         sig = inspect.signature(user_function)
         func_params = set(sig.parameters.keys())
-        expected_params = set(sim_params.keys()) | {"N"}
+        expected_params = set(sim_params.keys()) | {"N", "seed"}
 
         missing = expected_params - func_params
         if missing:
             raise Exception(
                 f"{user_function.__name__}() registration failed: "
                 f"Missing required parameters: {missing}. "
-                f"Function must accept all sim_params params plus 'N'."
+                f"Function must accept all sim_params params plus 'N' and 'seed'."
             )
 
-        extra = func_params - expected_params - {"seed"}
+        extra = func_params - expected_params
         if extra:
             raise Exception(
                 f"{user_function.__name__}() registration failed: "
                 f"Unexpected parameters: {extra}. "
-                f"Function must only accept sim_params params plus 'N' and optionally 'seed'."
+                f"Function must only accept sim_params params plus 'N' and 'seed'."
             )
 
         leap_binder.set_simulation(user_function, name, sim_params)
@@ -2175,6 +2175,8 @@ def tensorleap_status_table():
         row = _find_row("tensorleap simulation")
         if not row:
             return
+        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"]
         if failed:

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/code_loader/leaploader.py

@@ -16,14 +16,14 @@ from code_loader.contract.datasetclasses import DatasetSample, DatasetBaseHandle
     PreprocessResponse, VisualizerHandler, LeapData, \
     PredictionTypeHandler, MetadataHandler, CustomLayerHandler, MetricHandler, VisualizerHandlerData, MetricHandlerData, \
     MetricCallableReturnType, CustomLossHandlerData, CustomLossHandler, RawInputsForHeatmap, SamplePreprocessResponse, \
-    ElementInstance, custom_latent_space_attribute, DatasetIntegrationSetup, InstanceMetricHandler
+    ElementInstance, custom_latent_space_attribute, DatasetIntegrationSetup, InstanceMetricHandler, _simulation_context
 from code_loader.contract.enums import DataStateEnum, TestingSectionEnum, DataStateType, DatasetMetadataType
 from code_loader.contract.exceptions import DatasetScriptException
 from code_loader.contract.responsedataclasses import DatasetIntegParseResult, DatasetTestResultPayload, \
     DatasetPreprocess, DatasetSetup, DatasetInputInstance, DatasetOutputInstance, DatasetMetadataInstance, \
     VisualizerInstance, PredictionTypeInstance, ModelSetup, CustomLayerInstance, MetricInstance, CustomLossInstance, \
     EngineFileContract, SimulationInstance
-from code_loader.contract.sim_config import FloatBounds, IntBounds, CategoricalBounds
+from code_loader.contract.sim_config import FloatBounds, IntBounds, CategoricalBounds, sample_sim_params
 from code_loader.inner_leap_binder import global_leap_binder
 from code_loader.inner_leap_binder.leapbinder import mapping_runtime_mode_env_var_mame
 from code_loader.leaploaderbase import LeapLoaderBase
@@ -38,6 +38,7 @@ def _serialize_sim_bounds(bounds) -> dict:
     raise ValueError(f"Unknown bounds type: {type(bounds)}")
 
 
+
 class LeapLoader(LeapLoaderBase):
     def __init__(self, code_path: str, code_entry_name: str):
         super().__init__(code_path, code_entry_name)
@@ -223,6 +224,8 @@ class LeapLoader(LeapLoaderBase):
                 test_payloads.append(preprocess_test_payload)
                 handlers_test_payloads = self._check_handlers()
                 test_payloads.extend(handlers_test_payloads)
+                simulation_test_payloads = self._check_simulations()
+                test_payloads.extend(simulation_test_payloads)
                 is_valid = all([payload.is_passed for payload in test_payloads])
                 setup_response = self.get_dataset_setup_response(handlers_test_payloads)
 
@@ -299,6 +302,74 @@ class LeapLoader(LeapLoaderBase):
             result_payloads.extend(test_result)
         return result_payloads
 
+    def _check_simulations(self):
+        # type: () -> List[DatasetTestResultPayload]
+        result_payloads = []
+        for sim in global_leap_binder.setup_container.simulations:
+            test_result = DatasetTestResultPayload(sim.name)
+            try:
+                kwargs = sample_sim_params(sim.sim_config)
+                kwargs["N"] = 1
+                kwargs["seed"] = 0
+                _simulation_context["active"] = True
+                try:
+                    preprocess_response = sim.function(**kwargs)
+                finally:
+                    _simulation_context["active"] = False
+                if not isinstance(preprocess_response, PreprocessResponse):
+                    raise TypeError(
+                        "Expected PreprocessResponse, got {}".format(type(preprocess_response).__name__)
+                    )
+                preprocess_response.state = DataStateType.additional
+                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(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(
+                                handler.name
+                            )
+                        )
+            except Exception as e:
+                line_number, file_name, stacktrace = get_root_exception_file_and_line_number()
+                test_result.display[TestingSectionEnum.Errors.name] = (
+                    "{} in file {}, line_number: {}\nStacktrace:\n{}".format(
+                        repr(e), file_name, line_number, stacktrace
+                    )
+                )
+                test_result.is_passed = False
+            result_payloads.append(test_result)
+        return result_payloads
+
+    def run_simulation_check(self, sim_name):
+        # type: (str) -> Dict[str, npt.NDArray[np.float32]]
+        self.exec_script()
+        sim = next(
+            (s for s in global_leap_binder.setup_container.simulations if s.name == sim_name),
+            None,
+        )
+        if sim is None:
+            raise ValueError("No simulation registered with name '{}'".format(sim_name))
+        kwargs = sample_sim_params(sim.sim_config)
+        kwargs["N"] = 1
+        kwargs["seed"] = 0
+        _simulation_context["active"] = True
+        try:
+            sim_preprocess = sim.function(**kwargs)
+        finally:
+            _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(sim_preprocess.sample_ids[0], sim_preprocess)
+        return result
+
     @staticmethod
     def _get_all_dataset_base_handlers() -> List[Union[DatasetBaseHandler, MetadataHandler]]:
         all_dataset_base_handlers: List[Union[DatasetBaseHandler, MetadataHandler]] = []

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/code_loader/leaploaderbase.py

@@ -2,7 +2,7 @@
 
 from abc import abstractmethod
 
-from typing import Dict, List, Union, Type, Optional, Tuple
+from typing import Any, Dict, List, Union, Type, Optional, Tuple
 
 import numpy as np
 import numpy.typing as npt
@@ -153,6 +153,11 @@ class LeapLoaderBase:
     def get_heatmap_visualizer_raw_vis_input_arg_name(self, visualizer_name: str) -> Optional[str]:
         pass
 
+    @abstractmethod
+    def run_simulation_check(self, sim_name):
+        # type: (str) -> Dict[str, Any]
+        pass
+
     def is_custom_latent_space(self) -> bool:
         if not self.code_entry_name or not self.code_path:
             return False

{code_loader-1.0.180.dev5 → code_loader-1.0.181}/pyproject.toml

@@ -1,6 +1,7 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.180.dev5"
+
+version = "1.0.181"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"