Removed uneccsary files

lukaszgajewski · Jan 23, 2024 · 5ae9ab3 · 5ae9ab3
1 parent d0aef9c
commit 5ae9ab3
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 54 deletions.
diff --git a/mkdocs.yml b/mkdocs.yml
diff --git a/mkgendocs.yml b/mkgendocs.yml
diff --git a/src/model.py b/src/model.py
@@ -224,8 +224,11 @@ def remove_countries_except(self):
         keep = [
             country for country in self.trade_matrix.index if country in self.countries_to_keep
         ]
-        self.trade_matrix = self.trade_matrix.loc[keep, keep]
-        # TODO make this behaviour consistent with remove_countries
+        self.trade_matrix = self.trade_matrix.loc[keep, :]
+        # Now also remove the columns from the trade matrix which have no trade anymore
+        col_sums = self.trade_matrix.sum(axis=0)
+        b_filter = ~(col_sums.eq(0))
+        self.trade_matrix = self.trade_matrix.loc[:, b_filter]
 
         self.production_data = self.production_data.loc[keep]