datamint 2.18.3__tar.gz → 2.18.4__tar.gz

{datamint-2.18.3 → datamint-2.18.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datamint
-Version: 2.18.3
+Version: 2.18.4
 Summary: A library for interacting with the Datamint API, designed for efficient data management, processing and Deep Learning workflows.
 Requires-Python: >=3.10
 Classifier: Programming Language :: Python :: 3

{datamint-2.18.3 → datamint-2.18.4}/datamint/entities/resource.py

@@ -17,6 +17,7 @@ from .base_entity import BaseEntity, MISSING_FIELD
 from .cache_manager import CacheManager
 from datamint.api.base_api import BaseApi
 from datamint.types import CacheMode
+from medimgkit.nifti_utils import NIFTI_MIMES
 
 if TYPE_CHECKING:
     from datamint.api.endpoints.resources_api import ResourcesApi
@@ -153,7 +154,7 @@ class BaseResource(BaseEntity, ABC):
         Returns:
             True if the resource is a NIfTI file, False otherwise
         """
-        return self.mimetype in ('application/nifti', 'image/nifti')
+        return self.mimetype in NIFTI_MIMES
 
     def is_video(self) -> bool:
         """Check if the resource is a video file.

{datamint-2.18.3 → datamint-2.18.4}/datamint/entities/resources/dicom_resource.py

@@ -11,8 +11,8 @@ class DICOMResource(VolumeResource):
     resource_kind: ClassVar[str] = 'dicom'
     resource_priority: ClassVar[int] = 50
     storage_aliases: ClassVar[tuple[str, ...]] = ('DicomResource', 'DicomResourceHandler')
-    mimetypes: ClassVar[tuple[str, ...]] = ('application/dicom',)
-    filename_suffixes: ClassVar[tuple[str, ...]] = ('.dcm',)
+    mimetypes: ClassVar[tuple[str, ...]] = ('application/dicom', 'application/x-dicom')
+    filename_suffixes: ClassVar[tuple[str, ...]] = ('.dcm','.dicom')
 
     @property
     def uids(self) -> dict[str, str]:

{datamint-2.18.3 → datamint-2.18.4}/datamint/entities/resources/nifti_resource.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 from typing import ClassVar
 
 from .volume_resource import VolumeResource
+from medimgkit.nifti_utils import NIFTI_MIMES
 
 
 class NiftiResource(VolumeResource):
@@ -11,10 +12,10 @@ class NiftiResource(VolumeResource):
     resource_kind: ClassVar[str] = 'nifti'
     resource_priority: ClassVar[int] = 50
     storage_aliases: ClassVar[tuple[str, ...]] = ('NiftiResource', 'NiftiResourceHandler')
-    mimetypes: ClassVar[tuple[str, ...]] = ('application/nifti','application/x-nifti')
+    mimetypes: ClassVar[tuple[str, ...]] = tuple(NIFTI_MIMES)
     filename_suffixes: ClassVar[tuple[str, ...]] = ('.nii', '.nii.gz')
 
     @property
     def is_compressed(self) -> bool:
         """Whether the underlying NIfTI file is gzip-compressed."""
-        return self.filename.casefold().endswith('.nii.gz')
+        return self.filename.casefold().endswith('.nii.gz')

{datamint-2.18.3 → datamint-2.18.4}/datamint/mlflow/flavors/datamint_flavor.py

@@ -63,9 +63,8 @@ def _process_input_example(input_example: ModelInputExample | None) -> tuple[Mod
     }
     if input_example is None:
         import datetime
-        import json
 
-        input_resource = json.dumps(dict(
+        input_resource = dict(
             id='model_id',
             storage='DicomResource',
             filename='file.dcm',
@@ -75,7 +74,7 @@ def _process_input_example(input_example: ModelInputExample | None) -> tuple[Mod
             status='inbox',
             created_at=datetime.datetime.now().isoformat(),
             created_by='user@mail.com',
-            modality='CT')
+            modality='CT'
         )
         return [input_resource], datamint_params
     if not isinstance(input_example, tuple):

{datamint-2.18.3 → datamint-2.18.4}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "datamint"
 description = "A library for interacting with the Datamint API, designed for efficient data management, processing and Deep Learning workflows."
-version = "2.18.3"
+version = "2.18.4"
 dynamic = ["dependencies"]
 requires-python = ">=3.10"
 readme = "README.md"