Skip to content

Commit

Permalink
Merge branch 'fix/run_sca_job_fix_ruff' into 'main'
Browse files Browse the repository at this point in the history
fix run_sca job

See merge request es/ai/hannah/hannah!416
  • Loading branch information
L6REKcodw committed Oct 29, 2024
2 parents 8286034 + 8077ae6 commit 70d363d
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion hannah/callbacks/optimization.py
Original file line number Diff line number Diff line change
Expand Up @@ -94,7 +94,7 @@ def _add_monitor_mapping(self, monitor):
else:
self.directions.append(-1.0)

def on_train_batch_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule", outputs: 'STEP_OUTPUT', batch: Any, batch_idx: int) -> None:
def on_train_batch_end(self, trainer: "pl.Trainer", pl_module: "pl.LightningModule", outputs: 'STEP_OUTPUT', batch: Any, batch_idx: int) -> None: # noqa: F821
callback_metrics = trainer.callback_metrics

for k, v in callback_metrics.items():
Expand Down
2 changes: 1 addition & 1 deletion hannah/nas/spaces/mobilenet/mobilenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ def residual(input):


@dataflow
def add(input, other):
def add(input, other): # noqa: F811
return op("Add", input, other)


Expand Down

0 comments on commit 70d363d

Please sign in to comment.