Remove set_crs in tests

NOAA-OWP · Jul 23, 2024 · c84bd47 · c84bd47
1 parent 336a7c4
commit c84bd47
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/data_generation/data_tests.py b/tests/data_generation/data_tests.py
@@ -85,7 +85,7 @@ def generate_aligned_and_agreement_maps(
                     allow_benchmark_values=allow_benchmark_values,
                 )
 
-                agreement_map_computed.rio.set_crs(cam.rio.crs)
+                agreement_map_computed.rio.write_crs(cam.rio.crs, inplace=True)
 
                 agreement_map_computed.rio.set_nodata(-9999)
                 if np.nan in agreement_map_computed:

diff --git a/tests/test_visualization.py b/tests/test_visualization.py
@@ -11,7 +11,7 @@
     glob="categorical_plot_success",
 )
 def test_categorical_plot_success(candidate_map, crs, entries):
-    candidate_map.rio.set_crs(crs)
+    candidate_map.rio.write_crs(crs, inplace=True)
     viz_object = candidate_map.gval.cat_plot(basemap=None)
     assert len(viz_object.axes.get_legend().texts) == entries