Fix tests (#455)

* fix pytorch tests

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

* fix tests

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

* don't use formatting

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

* fix issue

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

* remove commented out code

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

* simplify change

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

---------

Signed-off-by: Tobias Pitters <tobias.pitters@gmail.com>

Author: CloseChoice

dice_ml/explainer_interfaces/dice_genetic.py

@@ -476,8 +476,11 @@ def find_counterfactuals(self, query_instance, desired_range, desired_class,
             if rest_members > 0:
                 new_generation_2 = np.zeros((rest_members, self.data_interface.number_of_features))
                 for new_gen_idx in range(rest_members):
-                    parent1 = random.choice(population[:int(len(population) / 2)])
-                    parent2 = random.choice(population[:int(len(population) / 2)])
+                    top_half = int(len(population) / 2)
+                    parent1_idx = random.randrange(top_half)
+                    parent2_idx = random.randrange(top_half)
+                    parent1 = population[parent1_idx]
+                    parent2 = population[parent2_idx]
                     child = self.mate(parent1, parent2, features_to_vary, query_instance)
                     new_generation_2[new_gen_idx] = child
 

dice_ml/model_interfaces/pytorch_model.py

@@ -22,9 +22,9 @@ def __init__(self, model=None, model_path='', backend='PYT', func=None, kw_args=
 
         super().__init__(model, model_path, backend, func, kw_args)
 
-    def load_model(self):
+    def load_model(self, weights_only=False):
         if self.model_path != '':
-            self.model = torch.load(self.model_path)
+            self.model = torch.load(self.model_path, weights_only=weights_only)
 
     def get_output(self, input_instance, model_score=True,
                    transform_data=False, out_tensor=False):
@@ -35,9 +35,9 @@ def get_output(self, input_instance, model_score=True,
         """
         input_tensor = input_instance
         if transform_data:
-            input_tensor = torch.tensor(self.transformer.transform(input_instance).to_numpy()).float()
+            input_tensor = torch.tensor(self.transformer.transform(input_instance).to_numpy(dtype=np.float64)).float()
         if not torch.is_tensor(input_instance):
-            input_tensor = torch.tensor(self.transformer.transform(input_instance).to_numpy()).float()
+            input_tensor = torch.tensor(self.transformer.transform(input_instance).to_numpy(dtype=np.float64)).float()
         out = self.model(input_tensor).float()
         if not out_tensor:
             out = out.data.numpy()