[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

hoss/adapter.py

@@ -35,14 +35,14 @@
 from pystac import Asset, Item
 
 from hoss.dimension_utilities import is_index_subset
+from hoss.exceptions import StagingFailed
 from hoss.harmony_log_context import set_logger
 from hoss.subset import subset_granule
 from hoss.utilities import (
     get_file_mimetype,
     raise_from_hoss_exception,
     unexecuted_url_requested,
 )
-from hoss.exceptions import StagingFailed
 
 
 class HossAdapter(BaseHarmonyAdapter):
@@ -189,11 +189,7 @@ def validate_message(self):
             if not hasattr(source, 'variables') or not source.variables:
                 self.logger.info('All variables will be retrieved.')
 
-    def hoss_stage(
-            self,
-            url: str,
-            asset_name: str,
-            output_mimetype: str) -> str:
+    def hoss_stage(self, url: str, asset_name: str, output_mimetype: str) -> str:
         """Stages the file to and S3 location and returns the url.
         Throws a retriable exception when there is a failure.
 
@@ -204,12 +200,10 @@ def hoss_stage(
                 asset_name,
                 output_mimetype,
                 location=self.message.stagingLocation,
-                logger=self.logger
+                logger=self.logger,
             )
 
             return output_url
 
         except Exception as exception:
-            raise StagingFailed(
-                'Staging failed, ' + str(exception)
-            ) from exception
+            raise StagingFailed('Staging failed, ' + str(exception)) from exception

tests/unit/test_adapter.py

@@ -563,12 +563,8 @@ def test_missing_variables(
             logger=hoss.logger,
         )
 
-    def test_hoss_stage(
-        self, mock_stage, mock_subset_granule, mock_get_mimetype
-    ):
-        """Ensure hoss_stage successfully calls hoss_stage and returns the URL
-
-        """
+    def test_hoss_stage(self, mock_stage, mock_subset_granule, mock_get_mimetype):
+        """Ensure hoss_stage successfully calls hoss_stage and returns the URL"""
         mock_subset_granule.return_value = '/path/to/output.nc'
         expected_output_basename = '/path/to/output.nc'
         mock_get_mimetype.return_value = ('application/x-netcdf4', None)
@@ -601,16 +597,13 @@ def test_hoss_stage(
             'africa.nc4',
             'application/x-netcdf4',
         )
-        
+
         assert result == expected_staged_url
 
     def test_hoss_stage_raises_exception(
-            self,
-            mock_stage,
-            mock_subset_granule,
-            mock_get_mimetype
-        ):
-        """Ensure hoss_stage raises an StagingFailed exception when 
+        self, mock_stage, mock_subset_granule, mock_get_mimetype
+    ):
+        """Ensure hoss_stage raises an StagingFailed exception when
         the Harmony stage function throws an exception.
 
         """
@@ -631,12 +624,9 @@ def test_hoss_stage_raises_exception(
                 '/path/to/output.nc',
                 'africa.nc4',
                 'application/x-netcdf4',
-        )
+            )
 
-        self.assertIsInstance(
-            context_manager.exception,
-            CustomError
-        )
+        self.assertIsInstance(context_manager.exception, CustomError)
         self.assertEqual(
             str(context_manager.exception),
             'Staging failed, Connection timeout',

tests/unit/test_utilities.py

@@ -446,7 +446,9 @@ def test_raise_from_hoss_exception(self):
                 raise_from_hoss_exception(failed_exception)
             self.assertNotIsInstance(context.exception, NoRetryException)
 
-        with self.subTest('UrlAccessFailedWithNoRetries (no-retry) raises NoRetryException.'):
+        with self.subTest(
+            'UrlAccessFailedWithNoRetries (no-retry) raises NoRetryException.'
+        ):
             failed_exception = UrlAccessFailedWithNoRetries(test_url, 400)
             with self.assertRaises(NoRetryException):
                 raise_from_hoss_exception(failed_exception)