Finalize UCI example

pomonam · Jun 26, 2024 · 7eabaf6 · 7eabaf6
1 parent 4bcc81d
commit 7eabaf6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/uci/pipeline.py b/examples/uci/pipeline.py
@@ -8,8 +8,7 @@
 from torch.utils.data import Dataset
 
 
-def construct_regression_mlp() -> nn.Module:
-    num_inputs = 8
+def construct_regression_mlp(num_inputs: int = 8) -> nn.Module:
     model = torch.nn.Sequential(
         nn.Linear(num_inputs, 128, bias=True),
         nn.ReLU(),