fixed warnings

Author: vaibhavbahel

src/baseline/v2/inference/docker/inference.py

@@ -1,4 +1,7 @@
 import argparse
+import time
+import warnings
+
 import torch
 import nibabel as nib
 from pathlib import Path
@@ -14,7 +17,8 @@
 )
 
 from models.unet import build_model
-import time 
+
+warnings.filterwarnings("ignore")
 
 # -----------------------------
 # ARGUMENTS

src/baseline/v2/inference/docker/requirments.txt

@@ -1,3 +1,4 @@
 monai==1.3.0
+torch==2.1.0
 nibabel
 tqdm