replaces incorrect column in extract script

Author: tomjemmett

model/data/databricks.py

@@ -138,9 +138,9 @@ def get_hsa_activity_table(self) -> pd.DataFrame:
         """
         return (
             self._spark.read.table("hsa_activity_tables")
-            .filter(F.col("provider") == self._dataset)
+            .filter(F.col("dataset") == self._dataset)
             .filter(F.col("fyear") == self._fyear)
-            .drop("provider", "fyear")
+            .drop("dataset", "fyear")
             .toPandas()
         )