remove extra variable

horovod · Aug 17, 2022 · 9ea4ecf · 9ea4ecf
1 parent 99b2ffb
commit 9ea4ecf
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/horovod/spark/lightning/remote.py b/horovod/spark/lightning/remote.py
@@ -52,10 +52,8 @@ def RemoteTrainer(estimator, metadata, ckpt_bytes, run_id, dataset_idx, train_ro
     terminate_on_nan = estimator.getTerminateOnNan()
     transformation_fn = estimator.getTransformationFn()
     transformation = transformation_fn if transformation_fn else None
-    edit_fields = estimator.getTransformationEditFields()
-    transformation_edit_fields = edit_fields if edit_fields else None
-    removed_fields = estimator.getTransformationRemovedFields()
-    transformation_removed_fields = removed_fields if removed_fields else None
+    transformation_edit_fields = estimator.getTransformationEditFields()
+    transformation_removed_fields = estimator.getTransformationRemovedFields()
     inmemory_cache_all = estimator.getInMemoryCacheAll()
     callbacks = estimator.getCallbacks() or []
     train_steps_per_epoch = estimator.getTrainStepsPerEpoch()