fixes #600

robertmartin8 · Dec 1, 2024 · e12417a · e12417a
1 parent d508869
commit e12417a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pypfopt/base_optimizer.py b/pypfopt/base_optimizer.py
@@ -106,7 +106,7 @@ def save_weights_to_file(self, filename="weights.csv"):
         """
         clean_weights = self.clean_weights()
 
-        ext = filename.split(".")[1]
+        ext = filename.split(".")[-1].lower()
         if ext == "csv":
             pd.Series(clean_weights).to_csv(filename, header=False)
         elif ext == "json":