From 5ecc3f802c8db5e706d747eca6aa00931a78985b Mon Sep 17 00:00:00 2001
From: 01Yan <140186842+01Yan@users.noreply.github.com>
Date: Fri, 29 Nov 2024 13:42:26 +0000
Subject: [PATCH] Filter unnecessary warnings

---
 PDMD/utils/_run.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/PDMD/utils/_run.py b/PDMD/utils/_run.py
index eff0d08..dcf0501 100644
--- a/PDMD/utils/_run.py
+++ b/PDMD/utils/_run.py
@@ -5,6 +5,7 @@
 from torch.utils.data import ConcatDataset
 import os.path as osp
 import os
+import warnings
 import numpy as np
 import torch
 import random
@@ -12,11 +13,11 @@
 from ..models import ENERGY_Model, FORCE_Model
 from . import get_timestring, MutilWaterDataset, split_dataset, worker_init_fn, train, val, draw_two_dimension, reverse_min_max_scaler_1d
 
-
 # os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
 # os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 # os.environ['CUDA_LAUNCH_BLOCKING'] = '0'
 # torch.set_default_dtype(torch.float64)
+warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is deprecated')
 
 def run(config):
     np.random.seed(config.seed)