Skip to content

Commit 96cd400

Browse files
Copilotmanujosephv
andcommitted
Fix devices_list type issue by changing devices type to Any
Co-authored-by: manujosephv <10508493+manujosephv@users.noreply.github.com>
1 parent 6160be1 commit 96cd400

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/pytorch_tabular/config/config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import os
77
import re
88
from dataclasses import MISSING, dataclass, field
9-
from typing import Any, Dict, Iterable, List, Optional
9+
from typing import Any, Dict, Iterable, List, Optional, Union
1010

1111
from omegaconf import OmegaConf
1212

@@ -400,7 +400,7 @@ class TrainerConfig:
400400
"choices": ["cpu", "gpu", "tpu", "ipu", "mps", "auto"],
401401
},
402402
)
403-
devices: Optional[int] = field(
403+
devices: Any = field(
404404
default=-1,
405405
metadata={
406406
"help": "Number of devices to train on. -1 uses all available devices."

0 commit comments

Comments
 (0)