fix: issue 155 (#158)

* fix: simple fix for #155

* fix: decimation factor allows floats

* fix: decimation int

Author: AngRodrigues

map2loop/project.py

@@ -357,6 +357,21 @@ def set_sampler(self, datatype: Datatype, sampler: Sampler):
             sampler (Sampler):
                 The sampler to use
         """
+        allowed_samplers = {
+            Datatype.STRUCTURE: SamplerDecimator,
+            Datatype.GEOLOGY: SamplerSpacing,
+            Datatype.FAULT: SamplerSpacing,
+            Datatype.FOLD: SamplerSpacing,
+            Datatype.DTM: SamplerSpacing,
+        }
+
+        # Check for wrong sampler
+        if datatype in allowed_samplers:
+            allowed_sampler_type = allowed_samplers[datatype]
+            if not isinstance(sampler, allowed_sampler_type):
+                raise ValueError(
+                    f"Got wrong argument for this datatype: {type(sampler).__name__}, please use {allowed_sampler_type.__name__} instead"
+                )
         ## does the enum print the number or the label?
         logger.info(f"Setting sampler for {datatype} to {sampler.sampler_label}")
         self.samplers[datatype] = sampler

map2loop/sampler.py

@@ -9,7 +9,7 @@
 import pandas
 import shapely
 import numpy
-from typing import Optional
+from typing import Optional, Union
 
 
 class Sampler(ABC):