fix: Remove name from element spec of PipeMode (#150)

zuoyuanh · Zuoyuan Huang · web-flow · commit 2013ad748470 · 2024-01-22T10:14:55.000-08:00
* Remove name from element spec of PipeMode

* Bump up docker client version in test env

---------

Co-authored-by: Zuoyuan Huang &lt;zuoyuan@amazon.com&gt;
diff --git a/src/sagemaker_tensorflow/pipemode.py b/src/sagemaker_tensorflow/pipemode.py
@@ -119,5 +119,4 @@ def element_spec(self):
         return tensor_spec.TensorSpec(
             shape=self.output_shapes,
             dtype=self.output_types,
-            name=self.channel,
         )
diff --git a/tox.ini b/tox.ini
@@ -52,7 +52,7 @@ deps =
     contextlib2==21.6.0
     tensorflow==2.13.0
     awslogs==0.14.0
-    docker==5.0.2
+    docker==6.1.0
     cmake==3.21.2
 
 [testenv:flake8]

Original file line number	Diff line number	Diff line change
`@@ -119,5 +119,4 @@ def element_spec(self):`
`119`	`119`	`return tensor_spec.TensorSpec(`
`120`	`120`	`shape=self.output_shapes,`
`121`	`121`	`dtype=self.output_types,`
`122`		`- name=self.channel,`
`123`	`122`	`)`