riteshghorse commented on code in PR #28564: URL: https://github.com/apache/beam/pull/28564#discussion_r1484374771
########## sdks/python/apache_beam/transforms/environments_test.py: ########## @@ -21,20 +21,40 @@ # pytype: skip-file import logging +import tempfile import unittest from apache_beam.options.pipeline_options import PortableOptions +from apache_beam.options.pipeline_options import SetupOptions from apache_beam.portability import common_urns from apache_beam.runners import pipeline_context +from apache_beam.runners.portability import stager from apache_beam.transforms import environments from apache_beam.transforms.environments import DockerEnvironment from apache_beam.transforms.environments import EmbeddedPythonEnvironment from apache_beam.transforms.environments import EmbeddedPythonGrpcEnvironment from apache_beam.transforms.environments import Environment from apache_beam.transforms.environments import ExternalEnvironment from apache_beam.transforms.environments import ProcessEnvironment +from apache_beam.transforms.environments import PyPIArtifactRegistry from apache_beam.transforms.environments import SubprocessSDKEnvironment +# create a temp directory so that all artifacts are put in the same directory. +tmp_dir = tempfile.mkdtemp() + + +def mock_python_sdk_dependencies(options, tmp_dir=tmp_dir): Review Comment: yeah I think this will look more clean, thanks! -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
