From dc6fa51db498aa47af91b98912b77923ab614cdb Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Sun, 13 Aug 2023 12:57:28 -0500
Subject: [PATCH] SSL4EO: fix sampling script (#1510)

---
 experiments/ssl4eo/sample_ssl4eo.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/experiments/ssl4eo/sample_ssl4eo.py b/experiments/ssl4eo/sample_ssl4eo.py
index 12d8ab08a..313978851 100755
--- a/experiments/ssl4eo/sample_ssl4eo.py
+++ b/experiments/ssl4eo/sample_ssl4eo.py
@@ -53,7 +53,7 @@ def get_world_cities(
     cols = ["city", "lat", "lng", "population"]
     cities = pd.read_csv(os.path.join(download_root, filename), usecols=cols)
     cities.at[8436, "population"] = 50789  # fix one bug (Tecax) in the csv file
-    cities = cities.nlargest("population", size)
+    cities = cities.nlargest(size, "population")
     return cities