diff --git a/composer/workflows/kubernetes_pod_operator.py b/composer/workflows/kubernetes_pod_operator.py index d1dcf2f4daf..6b282bd89e9 100644 --- a/composer/workflows/kubernetes_pod_operator.py +++ b/composer/workflows/kubernetes_pod_operator.py @@ -74,6 +74,9 @@ task_id='pod-ex-minimum', # Name of task you want to run, used to generate Pod ID. name='pod-ex-minimum', + # Entrypoint of the container, if not specified the Docker container's + # entrypoint is used. The cmds parameter is templated. + cmds=['echo'], # The namespace to run within Kubernetes, default namespace is # `default`. There is the potential for the resource starvation of # Airflow workers and scheduler within the Cloud Composer environment, diff --git a/composer/workflows/pythonvirtualenvoperator_python2_test.py b/composer/workflows/pythonvirtualenvoperator_python2_test.py index 0844dde5c56..5ec7c28082e 100644 --- a/composer/workflows/pythonvirtualenvoperator_python2_test.py +++ b/composer/workflows/pythonvirtualenvoperator_python2_test.py @@ -12,10 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -import sys - -import pytest - from . import unit_testing