Merge pull request #12 from alcole/data-bristol-demo

Merge pull request #11 from alcole/main
alcole · Sep 17, 2024 · 6d8c84d · 6d8c84d
2 parents 9233021 + e85fde6
commit 6d8c84d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,5 @@
 [![linting: pylint](https://img.shields.io/badge/linting-pylint-yellowgreen)](https://github.com/pylint-dev/pylint)
-![pylint](https://img.shields.io/badge/PyLint-5.71-orange?logo=python&logoColor=white)
+![pylint](https://img.shields.io/badge/PyLint-6.43-orange?logo=python&logoColor=white)
 # databricks-best-practices
 ![databricks logo](https://upload.wikimedia.org/wikipedia/commons/6/63/Databricks_Logo.png)
 

diff --git a/covid_analysis/transforms.py b/covid_analysis/transforms.py
@@ -21,4 +21,4 @@ def clean_spark_cols(pdf):
 # Convert index to column (works with pandas API on Spark, too).
 def index_to_col(df, colname):
     df[colname] = df.index
-    return df
+    return df
diff --git a/tests/transforms_test.py b/tests/transforms_test.py
@@ -37,7 +37,7 @@ def test_filter(raw_input_df):
 # The test data has NaNs for Daily ICU occupancy; this should get filled to 0.
 def test_pivot(raw_input_df):
   pivoted = pivot_and_clean(raw_input_df, 0)
-  assert pivoted["Daily ICU occupancy"][0] == 0
+  assert pivoted["Daily ICU occupancy"].iloc[0] == 0
 
 # Test column cleaning.
 def test_clean_cols(colnames_df):