Unverified Commit e403ce87 authored by Zhang, Chen's avatar Zhang, Chen Committed by GitHub
Browse files

Update src/imars3d/backend/corrections/ring_removal.py



use suggestion to make bm3dsr optional on import

Co-authored-by: default avatarPete Peterson <petersonpf@ornl.gov>
parent eaa7b7fa
Loading
Loading
Loading
Loading
+5 −1
Original line number Diff line number Diff line
@@ -6,7 +6,11 @@ import param
from imars3d.backend.util.functions import clamp_max_workers
import scipy
import numpy as np
try:
  import bm3d_streak_removal as bm3dsr
except ImportError:
  # add a log?
  bm3dsr = None
from multiprocessing.managers import SharedMemoryManager
from tqdm.contrib.concurrent import process_map
from functools import partial