Update bioimageio/core/build_spec/build_model.py

constantinpape · github-actions[bot] · web-flow · commit d3dd1a3d3143 · 2021-12-01T20:55:48.000+01:00
Co-authored-by: github-actions[bot] &lt;41898282+github-actions[bot]@users.noreply.github.com&gt;
diff --git a/bioimageio/core/build_spec/build_model.py b/bioimageio/core/build_spec/build_model.py
@@ -114,9 +114,7 @@ def _get_weights(
         )
 
     elif weight_type == "torchscript":
-        weights = model_spec.raw_nodes.TorchscriptWeightsEntry(
-            source=weight_source, sha256=weight_hash, **attachments
-        )
+        weights = model_spec.raw_nodes.TorchscriptWeightsEntry(source=weight_source, sha256=weight_hash, **attachments)
 
     elif weight_type == "keras_hdf5":
         if tensorflow_version is None:

Original file line number	Diff line number	Diff line change
`@@ -114,9 +114,7 @@ def _get_weights(`
`114`	`114`	`)`
`115`	`115`
`116`	`116`	`elif weight_type == "torchscript":`
`117`		`- weights = model_spec.raw_nodes.TorchscriptWeightsEntry(`
`118`		`- source=weight_source, sha256=weight_hash, **attachments`
`119`		`- )`
	`117`	`+ weights = model_spec.raw_nodes.TorchscriptWeightsEntry(source=weight_source, sha256=weight_hash, **attachments)`
`120`	`118`
`121`	`119`	`elif weight_type == "keras_hdf5":`
`122`	`120`	`if tensorflow_version is None:`