Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Nov 28, 2024
1 parent 5f58681 commit 75af2b7
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion apax/nodes/selection.py
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,9 @@ class BatchKernelSelection(zntrack.Node):
selected_ids: list[int] = zntrack.outs(independent=True)

models: typing.List[ApaxBase] = zntrack.deps()
base_feature_map: dict = zntrack.params(default_factory= lambda: {"name": "ll_grad", "layer_name": "dense_2"})
base_feature_map: dict = zntrack.params(
default_factory=lambda: {"name": "ll_grad", "layer_name": "dense_2"}
)
selection_method: str = zntrack.params("max_dist")
n_configurations: int = zntrack.params()
min_distance_threshold: int = zntrack.params(0.0)
Expand Down

0 comments on commit 75af2b7

Please sign in to comment.