Check checkpoint_options.checkpoint_dir: it may be unfilled in a caller's checkpoint_options.

dzelle · tensorflower-gardener · commit ff5e05efd35b · 2022-11-10T10:29:39.000-08:00
PiperOrigin-RevId: 487576430
diff --git a/tensorflow_gnn/runner/trainers/keras_fit.py b/tensorflow_gnn/runner/trainers/keras_fit.py
@@ -110,6 +110,8 @@ def __init__(
 
     if checkpoint_options is None:
       checkpoint_options = KerasTrainerCheckpointOptions()
+
+    if checkpoint_options.checkpoint_dir is None:
       checkpoint_options.checkpoint_dir = os.path.join(model_dir, "ckpnt")
 
     if backup_dir is None:
