Fix style, test, and extraneous import

neuralmagic · Sep 26, 2023 · d41d3d3 · d41d3d3
1 parent d6f01a0
commit d41d3d3
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 5 deletions.
diff --git a/src/sparsezoo/api/query_parser.py b/src/sparsezoo/api/query_parser.py
@@ -20,9 +20,7 @@
 
 DEFAULT_MODELS_FIELDS = ["modelId", "stub"]
 
-DEFAULT_FILES_FIELDS = [
-    "displayName", "downloadUrl", "fileSize", "fileType", "modelId"
-]
+DEFAULT_FILES_FIELDS = ["displayName", "downloadUrl", "fileSize", "fileType", "modelId"]
 
 DEFAULT_TRAINING_RESULTS_FIELDS = [
     "datasetName",

diff --git a/src/sparsezoo/model/utils.py b/src/sparsezoo/model/utils.py
@@ -31,7 +31,7 @@
     ValidationResult,
 )
 from sparsezoo.objects import Directory, File, NumpyDirectory, OnnxGz, Recipes
-from sparsezoo.utils import BASE_API_URL, convert_to_bool, save_numpy
+from sparsezoo.utils import convert_to_bool, save_numpy
 
 
 __all__ = [

diff --git a/tests/sparsezoo/api/test_query_parser.py b/tests/sparsezoo/api/test_query_parser.py
@@ -148,7 +148,7 @@
                 "fields": (
                     "modelId displayName benchmarkResults "
                     "{ batchSize deviceInfo numCores recordedUnits recordedValue } "
-                    "files { displayName fileSize modelId fileType } "
+                    "files { displayName downloadUrl fileSize fileType modelId } "
                 ),
             },
         ),