Merge pull request #223 from delira-dev/transform_typecheckfix

Fix Type Checks
delira-dev · Oct 7, 2019 · 47d2a14 · 47d2a14
2 parents 7ab64d6 + 90bd460
commit 47d2a14
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/delira/data_loading/data_manager.py b/delira/data_loading/data_manager.py
@@ -93,8 +93,8 @@ class defining the sampling strategy
 
         self.data = data
 
-        if not inspect.isclass(sampler_cls) and issubclass(sampler_cls,
-                                                           AbstractSampler):
+        if not (inspect.isclass(sampler_cls) and issubclass(sampler_cls,
+                                                            AbstractSampler)):
             raise TypeError
 
         self.sampler_cls = sampler_cls
@@ -302,8 +302,8 @@ def transforms(self, new_transforms):
 
         """
 
-        if new_transforms is not None or isinstance(new_transforms,
-                                                    AbstractTransform):
+        if new_transforms is not None and not isinstance(
+                new_transforms, AbstractTransform):
             raise TypeError
 
         self._transforms = new_transforms