Skip to content

Commit

Permalink
fixing mypy error with new end to end run
Browse files Browse the repository at this point in the history
  • Loading branch information
arik-shurygin committed Jan 7, 2025
1 parent e11e4fd commit 058a3a8
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions examples/example_end_to_end_run.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,19 +39,20 @@


class ExampleDynodeRunner(AbstractDynodeRunner):
def process_state(self, _: str, infer: bool = False):
def process_state(self, state: str, **kwargs):
"""An example of a method used to process a single state using DynODE
In this case the example configs are built around US data, so we are
running the whole US as a single entity.
Parameters
----------
_ : str
state : str
state USPS, ignored in this specific example
infer : bool, optional
whether or not the user of this example script wants to run inference,
by default False
"""
infer = bool(kwargs["infer"])
# step 1: define your paths
config_path = "examples/config/"
# global_config include definitions such as age bin bounds and strain definitions
Expand Down Expand Up @@ -153,4 +154,4 @@ def process_state(self, _: str, infer: bool = False):
os.mkdir("output")

runner = ExampleDynodeRunner("output/")
runner.process_state("USA", infer)
runner.process_state("USA", infer=infer)

0 comments on commit 058a3a8

Please sign in to comment.