Skip to content

Commit 8d515b0

Browse files
committed
Allow zero hidden layers for linear classifier.
1 parent 0bb15d1 commit 8d515b0

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

‎src/scunveil/_inference.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,10 @@ def _validate_checkpoint_metadata(self):
231231
value, (bool, np.bool_)
232232
):
233233
raise RuntimeError(f"config.{name} must be an integer.")
234-
if value <= 0:
234+
if name == "n_layers":
235+
if value < 0:
236+
raise RuntimeError("config.n_layers must be non-negative.")
237+
elif value <= 0:
235238
raise RuntimeError(f"config.{name} must be positive.")
236239

237240
required_columns = {"feature_id", "feature_name"}

0 commit comments

Comments
 (0)