feat: Publication Type Filter (#4)

* Filter for scholarly article and conference paper only
hicsail · Jun 25, 2024 · 2ad396a · 2ad396a
1 parent 10a2c1c
commit 2ad396a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/packages/frp/frp/scholarly.py b/packages/frp/frp/scholarly.py
@@ -68,6 +68,7 @@ def _filter(self, df: pd.DataFrame, year: int) -> pd.DataFrame:
         """
         # Filter for everything on the current year or later
         df = df[df['Reporting date 1'].dt.year >= year]
+        df = df[(df['Scholarly & creative work type'] == 'Scholarly article') | (df['Scholarly & creative work type'] == 'Conference paper (Published)')]
         return df
 
     def _augment(self, df: pd.DataFrame) -> pd.DataFrame: