Skip to content

Commit 7aca195

Browse files
Define train method are it is required - removed hardcode line in main (#8)
Signed-off-by: Andres <diazandr3s@gmail.com> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent f519b79 commit 7aca195

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

monailabel/monaivista/lib/configs/vista_point_2pt5.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,3 +165,6 @@ def infer(self) -> Union[InferTask, Dict[str, InferTask]]:
165165
preload=strtobool(self.conf.get("preload", "false")),
166166
)
167167
return task
168+
169+
def trainer(self) -> Optional[TrainTask]:
170+
None

monailabel/monaivista/main.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,9 +207,8 @@ def main():
207207
force=True,
208208
)
209209

210-
str(Path.home())
211-
# studies = f"{home}/Dataset/Radiology"
212-
studies = "/home/andres/Documents/workspace/disk-workspace/DatasetsMore/temp-TotalSegmentatorDataset/quickTest/"
210+
home = str(Path.home())
211+
studies = f"{home}/Dataset/Radiology/WholeBodyCTDataset/test"
213212

214213
parser = argparse.ArgumentParser()
215214
parser.add_argument("-s", "--studies", default=studies)

0 commit comments

Comments
 (0)