code-loader 1.0.93.dev2__tar.gz → 1.0.93.dev3__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of code-loader might be problematic. Click here for more details.

Files changed (32) hide show
  1. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/PKG-INFO +1 -1
  2. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/datasetclasses.py +1 -1
  3. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/inner_leap_binder/leapbinder_decorators.py +5 -2
  4. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/pyproject.toml +1 -1
  5. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/LICENSE +0 -0
  6. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/README.md +0 -0
  7. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/__init__.py +0 -0
  8. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/__init__.py +0 -0
  9. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/enums.py +0 -0
  10. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/exceptions.py +0 -0
  11. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/mapping.py +0 -0
  12. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/responsedataclasses.py +0 -0
  13. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/contract/visualizer_classes.py +0 -0
  14. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/default_losses.py +0 -0
  15. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/default_metrics.py +0 -0
  16. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/__init__.py +0 -0
  17. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/api.py +0 -0
  18. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/cli_config_utils.py +0 -0
  19. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/client.py +0 -0
  20. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/epoch.py +0 -0
  21. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/experiment.py +0 -0
  22. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/experiment_context.py +0 -0
  23. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/types.py +0 -0
  24. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/utils.py +0 -0
  25. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/experiment_api/workingspace_config_utils.py +0 -0
  26. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/inner_leap_binder/__init__.py +0 -0
  27. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/inner_leap_binder/leapbinder.py +0 -0
  28. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/leaploader.py +0 -0
  29. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/leaploaderbase.py +0 -0
  30. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/utils.py +0 -0
  31. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/visualizers/__init__.py +0 -0
  32. {code_loader-1.0.93.dev2 → code_loader-1.0.93.dev3}/code_loader/visualizers/default_visualizers.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 1.0.93.dev2
3
+ Version: 1.0.93.dev3
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
@@ -205,7 +205,7 @@ class MetadataHandler:
205
205
  class PredictionTypeHandler:
206
206
  name: str
207
207
  labels: List[str]
208
- channel_dim: int
208
+ channel_dim: int = -1
209
209
 
210
210
 
211
211
  @dataclass
@@ -8,7 +8,7 @@ import numpy.typing as npt
8
8
  from code_loader.contract.datasetclasses import CustomCallableInterfaceMultiArgs, \
9
9
  CustomMultipleReturnCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, CustomCallableInterface, \
10
10
  VisualizerCallableInterface, MetadataSectionCallableInterface, PreprocessResponse, SectionCallableInterface, \
11
- ConfusionMatrixElement, SamplePreprocessResponse
11
+ ConfusionMatrixElement, SamplePreprocessResponse, PredictionTypeHandler
12
12
  from code_loader.contract.enums import MetricDirection, LeapDataType, DatasetMetadataType
13
13
  from code_loader import leap_binder
14
14
  from code_loader.contract.mapping import NodeMapping, NodeMappingType, NodeConnection
@@ -35,7 +35,10 @@ def _add_mapping_connections(connects_to, arg_names, node_mapping_type, name):
35
35
 
36
36
 
37
37
 
38
- def tensorleap_load_model():
38
+ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = None):
39
+ for i, prediction_type in enumerate(prediction_types):
40
+ leap_binder.add_prediction(prediction_type.name, prediction_type.labels, prediction_type.channel_dim, i)
41
+
39
42
  def decorating_function(load_model_func):
40
43
  class TempMapping:
41
44
  pass
@@ -1,6 +1,6 @@
1
1
  [tool.poetry]
2
2
  name = "code-loader"
3
- version = "1.0.93.dev2"
3
+ version = "1.0.93.dev3"
4
4
  description = ""
5
5
  authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
6
6
  license = "MIT"