Skip to content

Commit

Permalink
chore: cleanup code styles
Browse files Browse the repository at this point in the history
  • Loading branch information
hellkite500 committed Jul 23, 2024
1 parent 8be54bd commit 81dae55
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions python/ngen_cal/src/ngen/cal/ngen_hooks/ngen_output.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,12 @@
if TYPE_CHECKING:
from ngen.cal.meta import JobMeta

class TrouteOutput():
from typing import TYPE_CHECKING

if TYPE_CHECKING:
from ngen.cal.meta import JobMeta

class TrouteOutput:

def __init__(self, filepath: Path) -> None:
self._output_file = filepath
Expand Down Expand Up @@ -38,7 +43,7 @@ def _get_dataframe(self) -> DataFrame | None:
# Try external provided output hooks, if those fail, try this one
# this will only execute if all other hooks return None (or they don't exist)
@hookimpl(specname="ngen_cal_model_output", trylast=True)
def getOutput(self, id: str) -> Series:
def get_output(self, id: str) -> Series:
try:
#look for routed data
#read the routed flow at the given id
Expand Down

0 comments on commit 81dae55

Please sign in to comment.