code-loader 1.0.93.dev2__py3-none-any.whl → 1.0.93.dev3__py3-none-any.whl

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 (6) hide show
  1. code_loader/contract/datasetclasses.py +1 -1
  2. code_loader/inner_leap_binder/leapbinder_decorators.py +5 -2
  3. {code_loader-1.0.93.dev2.dist-info → code_loader-1.0.93.dev3.dist-info}/METADATA +1 -1
  4. {code_loader-1.0.93.dev2.dist-info → code_loader-1.0.93.dev3.dist-info}/RECORD +6 -6
  5. {code_loader-1.0.93.dev2.dist-info → code_loader-1.0.93.dev3.dist-info}/LICENSE +0 -0
  6. {code_loader-1.0.93.dev2.dist-info → code_loader-1.0.93.dev3.dist-info}/WHEEL +0 -0
code_loader/contract/datasetclasses.py CHANGED
@@ -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
code_loader/inner_leap_binder/leapbinder_decorators.py CHANGED
@@ -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
{code_loader-1.0.93.dev2.dist-info → code_loader-1.0.93.dev3.dist-info}/METADATA RENAMED
@@ -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
{code_loader-1.0.93.dev2.dist-info → code_loader-1.0.93.dev3.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
2
2
  code_loader/__init__.py,sha256=6MMWr0ObOU7hkqQKgOqp4Zp3I28L7joGC9iCbQYtAJg,241
3
3
  code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
- code_loader/contract/datasetclasses.py,sha256=ZN6XC-tg0qaT-GF4YiV2SsM2dUQW4fc1akLF-J4YT0w,7964
4
+ code_loader/contract/datasetclasses.py,sha256=sNBHyfdg2uS3rL65kJVZjjNVbSXy0dTQMganldQjGNw,7969
5
5
  code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
6
6
  code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
7
7
  code_loader/contract/mapping.py,sha256=e11h_sprwOyE32PcqgRq9JvyahQrPzwqgkhmbQLKLQY,1165
@@ -21,13 +21,13 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
21
21
  code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
22
22
  code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
23
23
  code_loader/inner_leap_binder/leapbinder.py,sha256=mRg5ofWNbUkRsIafaz7i7NKi-1USgeAuNTIBxASSMcw,31713
24
- code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=TpAgM6VTdBN4yIFCZaPXxO4vB2QgUmeo0rICqBA_f-U,29764
24
+ code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=dsoAp7puFflisifdmR2JOtVF_3FvPy4quSr2kxgGGps,30022
25
25
  code_loader/leaploader.py,sha256=GGHZzovqGsM6oY14pwyL35LeOmdRfjVuhhY2ZqlYKaM,26481
26
26
  code_loader/leaploaderbase.py,sha256=VH0vddRmkqLtcDlYPCO7hfz1_VbKo43lUdHDAbd4iJc,4198
27
27
  code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
28
28
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
29
29
  code_loader/visualizers/default_visualizers.py,sha256=669lBpLISLO6my5Qcgn1FLDDeZgHumPf252m4KHY4YM,2555
30
- code_loader-1.0.93.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
31
- code_loader-1.0.93.dev2.dist-info/METADATA,sha256=MySowabFif2EsHj8v1oZzNOth38H2U_PKieGriC_pxQ,854
32
- code_loader-1.0.93.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
33
- code_loader-1.0.93.dev2.dist-info/RECORD,,
30
+ code_loader-1.0.93.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
31
+ code_loader-1.0.93.dev3.dist-info/METADATA,sha256=KE38NLhtwRUwok13eyAUfZOSELgk-BzlXCBzhKPbvOQ,854
32
+ code_loader-1.0.93.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
33
+ code_loader-1.0.93.dev3.dist-info/RECORD,,