From d41d3d3c21d7c10d20eaf71676130879004efdcb Mon Sep 17 00:00:00 2001 From: Danny Guinther Date: Tue, 26 Sep 2023 11:15:32 -0400 Subject: [PATCH] Fix style, test, and extraneous import --- src/sparsezoo/api/query_parser.py | 4 +--- src/sparsezoo/model/utils.py | 2 +- tests/sparsezoo/api/test_query_parser.py | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) diff --git a/src/sparsezoo/api/query_parser.py b/src/sparsezoo/api/query_parser.py index 9b124013..2e5994ac 100644 --- 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 index a8c9cbd9..966e605c 100644 --- 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 index dce7472e..82622799 100644 --- 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 } " ), }, ),