Reformat

src/cli/cli.py

@@ -129,10 +129,15 @@
     action="store_true",
 )
 arg_register_dataset.add_argument(
-    "--schema", default=f"{SCHEMA_VERSION}", help="Which schema to connect to",
+    "--schema",
+    default=f"{SCHEMA_VERSION}",
+    help="Which schema to connect to",
 )
 arg_register_dataset.add_argument(
-    "--locale", help="Location where dataset was produced", type=str, default="NERSC",
+    "--locale",
+    help="Location where dataset was produced",
+    type=str,
+    default="NERSC",
 )
 arg_register_dataset.add_argument(
     "--owner", help="Owner of dataset. Defaults to $USER."

src/cli/register.py

@@ -31,7 +31,7 @@ def register_dataset(args):
         execution_start=args.execution_start,
         execution_locale=args.execution_locale,
         execution_configuration=args.execution_configuration,
-        input_datasets=args.input_datasets
+        input_datasets=args.input_datasets,
     )
 
     print(f"Created dataset entry with id {new_id}")

src/dataregistry/DataRegistry.py

@@ -42,16 +42,17 @@ def __init__(
             True for more output.
         """
         # Establish connection to database
-        self.db_connection = DbConnection(
-            config_file, schema=schema, verbose=verbose
-        )
+        self.db_connection = DbConnection(config_file, schema=schema, verbose=verbose)
 
         # Create query object
         self.Query = Query(self.db_connection)
 
         # Create registrar object
         self.Registrar = Registrar(
-            self.db_connection, owner=owner, owner_type=owner_type, root_dir=root_dir,
+            self.db_connection,
+            owner=owner,
+            owner_type=owner_type,
+            root_dir=root_dir,
         )
 
         # Create query object

src/dataregistry/exceptions.py

@@ -7,7 +7,6 @@ class DataRegistryException(Exception):
 
 class DataRegistryNYI(DataRegistryException):
     def __init__(self, feature=""):
-
         msg = f"Feature {feature} not yet implemented"
         self.msg = msg
         super().__init__(self.msg)

src/dataregistry/query.py

@@ -272,7 +272,11 @@ def get_db_versioning(self):
         )
 
     def find_datasets(
-        self, property_names=None, filters=[], verbose=False, return_format="property_dict"
+        self,
+        property_names=None,
+        filters=[],
+        verbose=False,
+        return_format="property_dict",
     ):
         """
         Get specified properties for datasets satisfying all filters

src/dataregistry/registrar.py

@@ -29,7 +29,11 @@
 
 class Registrar:
     def __init__(
-        self, db_connection, owner=None, owner_type=None, root_dir=None,
+        self,
+        db_connection,
+        owner=None,
+        owner_type=None,
+        root_dir=None,
     ):
         """
         Class to register new datasets, executions and alias names.

src/dataregistry/registrar_util.py

@@ -181,17 +181,17 @@ def _name_from_relpath(relative_path):
     We use this when the dataset name is not explicitly defined, and we take it
     from the final directory if path.
 
-	e.g, /root/to/dataset/dir would return "dir"
+        e.g, /root/to/dataset/dir would return "dir"
 
-	Parameters
-	----------
-	relative_path : str
-		Path to dataset (can be relative or absolute)
+        Parameters
+        ----------
+        relative_path : str
+                Path to dataset (can be relative or absolute)
 
-	Returns
-	-------
-	name : str
-		Scraped name of dataset
+        Returns
+        -------
+        name : str
+                Scraped name of dataset
     """
 
     relpath = relative_path