typing in population_view.py (#626)

hussain-jafari · web-flow · commit 853242d597c1 · 2025-06-02T09:35:51.000-07:00
Category: typing JIRA issue: https://jira.ihme.washington.edu/browse/MIC-6107 Testing check builds
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,3 +1,7 @@
+**3.4.5 - 05/30/25**
+
+  - Type-hinting: Fix mypy errors in framework/population/population_view.py
+
 **3.4.4 - 05/27/25**
 
   - Feature: Update package pins for isort and pandas stubs
diff --git a/src/vivarium/framework/population/population_view.py b/src/vivarium/framework/population/population_view.py
@@ -499,4 +499,7 @@ def _update_column_and_ensure_dtype(
                 f"the {update.name} column from {existing.dtype} to {update.dtype}."
             )
         new_state_table_values = new_state_table_values.astype(update_values.dtype)
-        return pd.Series(new_state_table_values, index=existing.index, name=existing.name)
+        new_state_table: pd.Series[Any] = pd.Series(
+            new_state_table_values, index=existing.index, name=existing.name
+        )
+        return new_state_table

Original file line number	Diff line number	Diff line change
`@@ -499,4 +499,7 @@ def _update_column_and_ensure_dtype(`
`499`	`499`	`f"the {update.name} column from {existing.dtype} to {update.dtype}."`
`500`	`500`	`)`
`501`	`501`	`new_state_table_values = new_state_table_values.astype(update_values.dtype)`
`502`		`- return pd.Series(new_state_table_values, index=existing.index, name=existing.name)`
	`502`	`+ new_state_table: pd.Series[Any] = pd.Series(`
	`503`	`+ new_state_table_values, index=existing.index, name=existing.name`
	`504`	`+ )`
	`505`	`+ return new_state_table`