update manifest flow

wandb · Jan 14, 2021 · 6c7987a · 6c7987a
1 parent 6955722
commit 6c7987a
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 10 deletions.
diff --git a/wandb/sdk/internal/internal_api.py b/wandb/sdk/internal/internal_api.py
@@ -1807,7 +1807,7 @@ def create_artifact_manifest(
             $projectName: String!,
             $runName: String!,
             $includeUpload: Boolean!,
-            $type: ArtifactManifestType! = FULL
+            $type: ArtifactManifestType = FULL
         ) {
             createArtifactManifest(input: {
                 name: $name,
@@ -1837,6 +1837,7 @@ def create_artifact_manifest(
         entity_name = entity or self.settings("entity")
         project_name = project or self.settings("project")
         run_name = run or self.current_run_id
+        type = type or "FULL"
 
         response = self.gql(
             mutation,

diff --git a/wandb/sdk_py27/internal/artifacts.py b/wandb/sdk_py27/internal/artifacts.py
@@ -138,10 +138,7 @@ def before_commit():
                 path = os.path.abspath(fp.name)
                 json.dump(self._manifest.to_manifest_json(), fp, indent=4)
             digest = wandb.util.md5_file(path)
-            _, resp = self._api.update_artifact_manifest(
-                manifest_id,
-                digest=digest,
-            )
+            _, resp = self._api.update_artifact_manifest(manifest_id, digest=digest,)
 
             # We're duplicating the file upload logic a little, which isn't great.
             upload_url = resp["uploadUrl"]

diff --git a/wandb/sdk_py27/internal/internal_api.py b/wandb/sdk_py27/internal/internal_api.py
@@ -1807,7 +1807,7 @@ def create_artifact_manifest(
             $projectName: String!,
             $runName: String!,
             $includeUpload: Boolean!,
-            $type: ArtifactManifestType! = FULL
+            $type: ArtifactManifestType = FULL
         ) {
             createArtifactManifest(input: {
                 name: $name,
@@ -1837,6 +1837,7 @@ def create_artifact_manifest(
         entity_name = entity or self.settings("entity")
         project_name = project or self.settings("project")
         run_name = run or self.current_run_id
+        type = type or "FULL"
 
         response = self.gql(
             mutation,
@@ -1853,8 +1854,10 @@ def create_artifact_manifest(
             },
         )
 
-        return response["createArtifactManifest"]["artifactManifest"]["id"],\
-               response["createArtifactManifest"]["artifactManifest"]["file"]
+        return (
+            response["createArtifactManifest"]["artifactManifest"]["id"],
+            response["createArtifactManifest"]["artifactManifest"]["file"],
+        )
 
     def update_artifact_manifest(
         self,
@@ -1901,8 +1904,10 @@ def update_artifact_manifest(
             },
         )
 
-        return response["updateArtifactManifest"]["artifactManifest"]["id"],\
-               response["updateArtifactManifest"]["artifactManifest"]["file"]
+        return (
+            response["updateArtifactManifest"]["artifactManifest"]["id"],
+            response["updateArtifactManifest"]["artifactManifest"]["file"],
+        )
 
     @normalize_exceptions
     def create_artifact_files(self, artifact_files):