Skip to content

Commit 7519989

Browse files
juanuribe28Tensorflow Cloud maintainers
authored andcommitted
Fix error when distribute_strategy and entry_point are None
PiperOrigin-RevId: 383453586
1 parent 7593fe4 commit 7519989

File tree

1 file changed

+2
-2
lines changed
  • src/python/tensorflow_cloud/core

1 file changed

+2
-2
lines changed

src/python/tensorflow_cloud/core/run.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -264,8 +264,8 @@ def run(
264264
# This contains the `entry_point` wrapped in a distribution strategy.
265265
preprocessed_entry_point = None
266266
if (distribution_strategy == "auto"
267-
or entry_point.endswith("ipynb")
268-
or entry_point is None):
267+
or entry_point is None
268+
or entry_point.endswith("ipynb")):
269269
preprocessed_entry_point, \
270270
pep_file_descriptor = preprocess.get_preprocessed_entry_point(
271271
entry_point,

0 commit comments

Comments
 (0)