From 3661e092177fb192f3d90ac8815e5a6067cc6de9 Mon Sep 17 00:00:00 2001
From: neelkanthavvari <109115881+neelkanthavvari@users.noreply.github.com>
Date: Sun, 16 Apr 2023 17:13:30 -0400
Subject: [PATCH] Update main.py

---
 geofree/main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/geofree/main.py b/geofree/main.py
index 80e8b77..f5938a7 100644
--- a/geofree/main.py
+++ b/geofree/main.py
@@ -9,7 +9,7 @@
 from pytorch_lightning import seed_everything
 from pytorch_lightning.trainer import Trainer
 from pytorch_lightning.callbacks import ModelCheckpoint, Callback, LearningRateMonitor
-from pytorch_lightning.utilities.distributed import rank_zero_only
+from pytorch_lightning.utilities.rank_zero import rank_zero_only
 
 
 def get_obj_from_str(string, reload=False):