Fixed pytests

LiamCPinchbeck · Sep 4, 2024 · 17a1301 · 17a1301
1 parent 4d9efa0
commit 17a1301
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/gammabayes/dark_matter/spectral_models/core/csv_to_dict_class.py b/gammabayes/dark_matter/spectral_models/core/csv_to_dict_class.py
@@ -36,7 +36,7 @@ def _convert_numeric_columns(self, numeric_type):
             # Attempt to convert each column to the specified numeric type.
             # If a column contains non-numeric data, it will remain unchanged.
             self.df[column] = pd.to_numeric(self.df[column], errors='ignore', downcast='float')
-            if self.df[column].dtype == np.number or self.df[column].dtype == np.float_:
+            if self.df[column].dtype == np.number or self.df[column].dtype == float:
                 self.df[column] = self.df[column].astype(numeric_type)
 
 

diff --git a/gammabayes/likelihoods/irfs/fov_irf_slicer.py b/gammabayes/likelihoods/irfs/fov_irf_slicer.py
@@ -339,10 +339,10 @@ def _refresh_buffer_window(self):
         else:
             lat_stop = -(default_lat_vals[1] - lat_offset)
 
-        if lon_stop is 0:
+        if lon_stop == 0:
             lon_stop = None
 
-        if lat_stop is 0:
+        if lat_stop == 0:
             lat_stop = None
 
         self.buffer_window = (

diff --git a/tests/test_3_irfs.py b/tests/test_3_irfs.py
@@ -17,7 +17,7 @@ def test_irf_input_output():
 
     irf_loglike = IRF_LogLikelihood(axes=[energy_recon_axis, longitudeaxis, latitudeaxis], 
                                     dependent_axes = [energy_true_axis, longitudeaxistrue, latitudeaxistrue],
-                                    pointing_direction=np.asarray([10,10]).T, 
+                                    pointing_dir=np.asarray([10,10])*u.deg, 
                                     zenith=40, 
                                     hemisphere='North', 
                                     prod_vers=5)