Merge pull request GoogleCloudPlatform#165 from GoogleCloudPlatform/ajayhemnani-patch-3

ajayhemnani · web-flow · commit 72a11c312281 · 2023-03-13T12:10:54.000-07:00
Update runner.py
diff --git a/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/labs/lab-02.ipynb b/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/labs/lab-02.ipynb
@@ -56,7 +56,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib"
+    "%pip install --upgrade --user google-api-python-client google-auth-httplib2 google-auth-oauthlib"
    ]
   },
   {
diff --git a/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/labs/pipeline/runner.py b/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/labs/pipeline/runner.py
@@ -47,7 +47,7 @@
 
   beam_tmp_folder = '{}/beam/tmp'.format(Config.ARTIFACT_STORE_URI)
   beam_pipeline_args = [
-      '--runner=DataflowRunner',
+      '--runner=DirectRunner',
       '--experiments=shuffle_mode=auto',
       '--project=' + Config.PROJECT_ID,
       '--temp_location=' + beam_tmp_folder,
diff --git a/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/solutions/lab-02.ipynb b/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/solutions/lab-02.ipynb
@@ -56,7 +56,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib"
+    "%pip install --upgrade --user google-api-python-client google-auth-httplib2 google-auth-oauthlib"
    ]
   },
   {
diff --git a/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/solutions/pipeline/runner.py b/workshops/tfx-caip-tf23/lab-02-tfx-pipeline/solutions/pipeline/runner.py
@@ -47,7 +47,7 @@
 
   beam_tmp_folder = '{}/beam/tmp'.format(Config.ARTIFACT_STORE_URI)
   beam_pipeline_args = [
-      '--runner=DataflowRunner',
+      '--runner=DirectRunner',
       '--experiments=shuffle_mode=auto',
       '--project=' + Config.PROJECT_ID,
       '--temp_location=' + beam_tmp_folder,

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@`
`56`	`56`	`"metadata": {},`
`57`	`57`	`"outputs": [],`
`58`	`58`	`"source": [`
`59`		`- "%pip install --upgrade google-api-python-client google-auth-httplib2 google-auth-oauthlib"`
	`59`	`+ "%pip install --upgrade --user google-api-python-client google-auth-httplib2 google-auth-oauthlib"`
`60`	`60`	`]`
`61`	`61`	`},`
`62`	`62`	`{`