konfai 1.2.0__py3-none-any.whl → 1.2.1__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 konfai might be problematic. Click here for more details.

konfai/network/network.py CHANGED
@@ -6,7 +6,8 @@ from collections import OrderedDict
6
6
  from collections.abc import Callable, Iterable, Iterator, Sequence
7
7
  from enum import Enum
8
8
  from functools import partial
9
- from typing import Any, Self
9
+ from typing import Any
10
+ from typing_extensions import Self
10
11
 
11
12
  import numpy as np
12
13
  import torch
konfai/predictor.py CHANGED
@@ -715,7 +715,7 @@ class Predictor(DistributedObject):
715
715
  path = models_directory() + self.name + "/StateDict/"
716
716
  name = sorted(os.listdir(path))[-1]
717
717
  if os.path.exists(path + name):
718
- state_dicts.append(torch.load(path + name, weights_only=True))
718
+ state_dicts.append(torch.load(path + name, weights_only=False))
719
719
  else:
720
720
  raise Exception(f"Model : {path + name} does not exist !")
721
721
  return state_dicts
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: konfai
3
- Version: 1.2.0
3
+ Version: 1.2.1
4
4
  Summary: Modular and configurable Deep Learning framework with YAML and PyTorch
5
5
  Author-email: Valentin Boussot <boussot.v@gmail.com>
6
6
  License-Expression: Apache-2.0
@@ -1,7 +1,7 @@
1
1
  konfai/__init__.py,sha256=qjE9Rqxo1sMrkqGS8I5xlGQMZnjIfU-CGgSI5Wmbmbs,1231
2
2
  konfai/evaluator.py,sha256=xAKWUDvdSxqYRUsKqH6ieQF06LWa785aE4zLv4I3_i4,17850
3
3
  konfai/main.py,sha256=Fc4HcJEhPmgunj_f-QYyvQNvjHrKHSUv27Okgu6V5_A,3842
4
- konfai/predictor.py,sha256=a8QbZPMKrf8bFz1XKZs7j6HT0RXC09HdiIQbvUBqKBM,34597
4
+ konfai/predictor.py,sha256=-ZcHrFnP7fOUZ4SK4DpNYbir7iScWQnSOZfmeSLtg1I,34598
5
5
  konfai/trainer.py,sha256=4mc-8r-FxtX_EAn2su8qd-BLLQ0D0So8hh5rmuf6Hqs,27163
6
6
  konfai/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
7
7
  konfai/data/augmentation.py,sha256=vcJE7mosvUkwbpbTN_lGP0S1uJrJYGjlLrt9VnDdJYY,27792
@@ -24,15 +24,15 @@ konfai/models/segmentation/NestedUNet.py,sha256=W4uauwF0HY8ybi49iYiTlKLdJEyD7SaC
24
24
  konfai/models/segmentation/UNet.py,sha256=Pu_LiQdO4Mrzyn0HRE6rwxUjHGH4OG-JpzWB_U1K46g,5602
25
25
  konfai/network/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
26
26
  konfai/network/blocks.py,sha256=l70_oOcz5Hmol2xmxruG0kke_2SVgO3rXYXVTMSdAS8,15645
27
- konfai/network/network.py,sha256=PWI6W4sz7G5Pbb-79l7mL61AoyHWlQytRtBzvwh3Ro0,54800
27
+ konfai/network/network.py,sha256=0G3goIopB_r4PHj4ohOkbov344mVYxZjq5PH57QETmc,54829
28
28
  konfai/utils/ITK.py,sha256=HVed4Z96X1jTaWrrQNdoBMqOtVK9InAPlDBJu-5uv3g,15476
29
29
  konfai/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
30
30
  konfai/utils/config.py,sha256=a7t44CYMUT5oCDdjL94IswhCVfFbQ5FCgDWZktDDkc4,14347
31
31
  konfai/utils/dataset.py,sha256=Au22fcADKyDJMfS8Z9q8kEXLtKkoufJsH7Pwly6pALo,28288
32
32
  konfai/utils/utils.py,sha256=jCj3tZ8agQYceSY_tlVYp88UFPE5oUn6tXrqnZGrKiI,28410
33
- konfai-1.2.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
- konfai-1.2.0.dist-info/METADATA,sha256=0i5f81WQyblGv3hF9L4l-skjcPmx1NK7NwtPmcdAQIY,2451
35
- konfai-1.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
- konfai-1.2.0.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
37
- konfai-1.2.0.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
38
- konfai-1.2.0.dist-info/RECORD,,
33
+ konfai-1.2.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
34
+ konfai-1.2.1.dist-info/METADATA,sha256=UJzi2uPo_JMW4VeEWYm3IB0ZdEaIHbIrdDj4zNNw3WQ,2451
35
+ konfai-1.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
36
+ konfai-1.2.1.dist-info/entry_points.txt,sha256=fG82HRN5-g39ACSOCtij_I3N6EHxfYnMR0D7TI_8pW8,81
37
+ konfai-1.2.1.dist-info/top_level.txt,sha256=xF470dkIlFoFqTZEOlRehKJr4WU_8OKGXrJqYm9vWKs,7
38
+ konfai-1.2.1.dist-info/RECORD,,
File without changes