Fixt style

domino_data/datasets.py

@@ -9,9 +9,9 @@
 import backoff
 import httpx
 import urllib3
-from domino_data.data_sources import DataSourceClient, ObjectStoreDatasource
 
 import domino_data.configuration_gen
+from domino_data.data_sources import DataSourceClient, ObjectStoreDatasource
 
 from .auth import AuthenticatedClient, get_jwt_token
 from .logging import logger
@@ -324,7 +324,7 @@ def list_files(
             prefix=prefix,
             page_size=page_size,
             config={},
-            credential={}
+            credential={},
         )
 
     @backoff.on_exception(backoff.expo, UnauthenticatedError, max_time=60)
@@ -348,4 +348,10 @@ def get_file_url(  # pylint: disable=too-many-arguments
         """
         logger.info("get_file_url", dataset_unique_name=dataset_unique_name, file_name=file_name)
 
-        return self.datasource_client.get_key_url(datasource_id=dataset_unique_name, object_key=file_name, is_read_write=False, config={}, credential={})
+        return self.datasource_client.get_key_url(
+            datasource_id=dataset_unique_name,
+            object_key=file_name,
+            is_read_write=False,
+            config={},
+            credential={},
+        )

tests/test_dataset.py

@@ -10,6 +10,7 @@
 from domino_data import configuration_gen as ds_gen
 from domino_data import datasets as ds
 
+
 # Get Dataset
 @pytest.mark.vcr
 def test_get_dataset():
@@ -19,6 +20,7 @@ def test_get_dataset():
 
     assert isinstance(dataset_test, ds.Dataset)
 
+
 @pytest.mark.vcr
 def test_get_dataset_with_jwt(monkeypatch):
     """Client can fetch a dataset using JWT."""
@@ -30,6 +32,7 @@ def test_get_dataset_with_jwt(monkeypatch):
 
     client.get_dataset("dataset-test")
 
+
 @pytest.mark.vcr
 def test_get_dataset_does_not_exist():
     """Client raises an error when dataset does not exist."""
@@ -105,6 +108,7 @@ def test_get_file_url():
 
     assert "aFile" in url
 
+
 @pytest.mark.vcr
 def test_get_file_url_not_found():
     """Client gets an error when getting URL for a dataset that does not exist."""