From 90d5d7481db7ea9510f7a1df0d03dc1e5ed3a957 Mon Sep 17 00:00:00 2001 From: uceshaf Date: Sun, 5 Jan 2025 17:43:38 +0500 Subject: [PATCH] Fix missing dependencies in setup.py and resolve indentation error --- setup.py | 2 ++ src/embedder/models/SSL4EO_S2L1C.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 55b9e2b..72296db 100644 --- a/setup.py +++ b/setup.py @@ -11,6 +11,8 @@ install_requires=[ "torch>=1.10.0", "torchvision", + "torchgeo", # Added New + "open_clip_torch", # Added New "pandas", "geopandas", "numpy", diff --git a/src/embedder/models/SSL4EO_S2L1C.py b/src/embedder/models/SSL4EO_S2L1C.py index fad28dd..faea563 100644 --- a/src/embedder/models/SSL4EO_S2L1C.py +++ b/src/embedder/models/SSL4EO_S2L1C.py @@ -80,7 +80,7 @@ def preprocess(self, input): Returns: torch.Tensor: Preprocessed input, scaled by a factor of 10,000. """ - return input / 1e4 + return input / 1e4 def forward(self, input): """