HRRR - FileLoader - Rename parameter to from output_dir to input_dir

Rename this parameter since we are reading and not writing files.
USDA-ARS-NWRC · Apr 23, 2021 · 14031ef · 14031ef
1 parent 7e3826e
commit 14031ef
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/tests/data/hrrr/test_file_loader.py b/tests/data/hrrr/test_file_loader.py
@@ -52,7 +52,7 @@ def setUpClass(cls):
         cls.subject = FileLoader()
 
     def test_parameters(self, _data_patch, _df_patch):
-        self.subject.get_saved_data(*self.METHOD_ARGS, output_dir='path')
+        self.subject.get_saved_data(*self.METHOD_ARGS, input_dir='path')
 
         self.assertEqual(self.START_DATE, self.subject.start_date)
         self.assertEqual(self.END_DATE, self.subject.end_date)

diff --git a/tests/data/hrrr/test_hrrr_gold.py b/tests/data/hrrr/test_hrrr_gold.py
@@ -37,7 +37,7 @@ def testHRRRGribLoad(self):
             self.END_DATE,
             self.BBOX,
             file_type='grib2',
-            output_dir=self.hrrr_dir.as_posix(),
+            input_dir=self.hrrr_dir.as_posix(),
             force_zone_number=self.UTM_ZONE_NUMBER
         )
 

diff --git a/tests/test_hrrr_preprocessor.py b/tests/test_hrrr_preprocessor.py
@@ -57,7 +57,7 @@ def test_pre_process(self):
             pd.to_datetime('2018-07-22 03:00'),
             self.BBOX,
             file_type='grib2',
-            output_dir=self.output_path.as_posix(),
+            input_dir=self.output_path.as_posix(),
             force_zone_number=self.UTM_ZONE_NUMBER)
 
         self.assertCountEqual(

diff --git a/weather_forecast_retrieval/data/hrrr/file_loader.py b/weather_forecast_retrieval/data/hrrr/file_loader.py
@@ -50,7 +50,7 @@ def file_type(self):
 
     def get_saved_data(self,
                        start_date, end_date, bbox,
-                       output_dir=None, file_type='grib2',
+                       input_dir=None, file_type='grib2',
                        force_zone_number=None,
                        var_keys=None):
         """
@@ -61,7 +61,7 @@ def get_saved_data(self,
             start_date:     datetime for the start
             end_date:       datetime for the end
             bbox:           list of  [lonmin,latmin,lonmax,latmax]
-            output_dir:     Base path to location of files
+            input_dir:      Base path to location of files
             file_type:      'grib' or 'netcdf', determines how to read the file
             force_zone_number: UTM zone number to convert datetime to
             var_keys:       which keys to grab from smrf variables,
@@ -98,8 +98,8 @@ def get_saved_data(self,
 
         self.force_zone_number = force_zone_number
 
-        if output_dir is not None:
-            self.output_dir = output_dir
+        if input_dir is not None:
+            self.output_dir = input_dir
 
         self.log.info('Getting saved data')
         self.get_data(var_map)