fix broken import (#484)

CHANGELOG.rst

@@ -1,3 +1,7 @@
+**3.0.9 - 09/12/24**
+
+ - Fix broken import in scale_up module
+
 **3.0.8 - 09/12/24**
 
  - Fix broken tests in RiskAttributableDisease due to changes in how to set population_view

src/vivarium_public_health/treatment/scale_up.py

@@ -13,8 +13,8 @@
 from vivarium import Component
 from vivarium.framework.engine import Builder
 from vivarium.framework.lookup import LookupTable
-from vivarium.framework.time import Time
 from vivarium.framework.values import Pipeline
+from vivarium.types import Time
 
 from vivarium_public_health.utilities import EntityString
 

tests/treatment/test_scale_up.py

@@ -0,0 +1,7 @@
+from vivarium_public_health.treatment import LinearScaleUp
+
+
+def test_linear_scale_up_instantiation():
+    scale_up = LinearScaleUp("treatment.sqlns")
+
+    assert scale_up.treatment == "treatment.sqlns"