Skip to content

Commit

Permalink
Cover to bool and log it.
Browse files Browse the repository at this point in the history
  • Loading branch information
loucerac committed May 5, 2023
1 parent a150ef7 commit a9a811c
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 0 deletions.
2 changes: 2 additions & 0 deletions drexml/cli/stab_explainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,6 +84,8 @@
n_chunks = max(1, n_devices)
chunk_size = len(features_val) // (n_chunks) + 1

print(f"{add=}")

def runner(model, bkg, new, check_add, use_gpu):

gpu_id = queue.get()
Expand Down
1 change: 1 addition & 0 deletions drexml/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,6 +73,7 @@ def parse_stab(argv):
n_gpus = int(n_gpus)
n_cpus = int(n_cpus)
debug = bool(int(debug))
add = bool(int(add))

if mode == "final":
n_splits = 1
Expand Down

0 comments on commit a9a811c

Please sign in to comment.