Commit d41d9035 authored by BO ZHANG's avatar BO ZHANG 🏀
Browse files

add filter_warnings

parent 0e355e6d
import json import json
import os.path import os.path
import subprocess import subprocess
import warnings
from astropy import time from astropy import time
...@@ -22,6 +23,7 @@ class Pipeline: ...@@ -22,6 +23,7 @@ class Pipeline:
n_jobs_cpu=18, n_jobs_cpu=18,
n_jobs_gpu=9, n_jobs_gpu=9,
device="CPU", device="CPU",
filter_warnings=False,
): ):
self.dir_in = dir_in self.dir_in = dir_in
self.dir_out = dir_out self.dir_out = dir_out
...@@ -54,6 +56,9 @@ class Pipeline: ...@@ -54,6 +56,9 @@ class Pipeline:
self.dfs = DFS(n_try=5) self.dfs = DFS(n_try=5)
self.crds = CRDS() self.crds = CRDS()
if filter_warnings:
self.filter_warnings()
def set_test_env(self): def set_test_env(self):
os.environ["CSST_DFS_API_MODE"] = "cluster" os.environ["CSST_DFS_API_MODE"] = "cluster"
os.environ["CSST_DFS_GATEWAY"] = "172.24.27.2:30880" os.environ["CSST_DFS_GATEWAY"] = "172.24.27.2:30880"
...@@ -71,6 +76,16 @@ class Pipeline: ...@@ -71,6 +76,16 @@ class Pipeline:
def now(self): def now(self):
return time.Time.now().isot return time.Time.now().isot
@staticmethod
def filter_warnings():
# Suppress all warnings
warnings.filterwarnings("ignore")
@staticmethod
def reset_warnings(self):
# Reset warning filters
warnings.resetwarnings()
class Message: class Message:
def __init__(self, file_path=""): def __init__(self, file_path=""):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment