diff --git a/CHANGELOG.md b/CHANGELOG.md index bc60b6f1c8..d702c11762 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -31,6 +31,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0 ### Removed +- Pickle dependency for CorrDiff. + ### Fixed - Consistent handling of single GPU runs in DistributedManager diff --git a/modulus/utils/generative/__init__.py b/modulus/utils/generative/__init__.py index e9bdcb9b20..a848c33bb8 100644 --- a/modulus/utils/generative/__init__.py +++ b/modulus/utils/generative/__init__.py @@ -36,7 +36,6 @@ get_obj_by_name, get_obj_from_module, get_top_level_function_name, - is_pickleable, is_top_level_function, list_dir_recursively_with_ignore, named_params_and_buffers, diff --git a/modulus/utils/generative/utils.py b/modulus/utils/generative/utils.py index 4825b29fe2..46e6829c44 100644 --- a/modulus/utils/generative/utils.py +++ b/modulus/utils/generative/utils.py @@ -23,9 +23,7 @@ import fnmatch import importlib import inspect -import io import os -import pickle import re import shutil import sys @@ -209,15 +207,6 @@ def get_dtype_and_ctype(type_obj: Any) -> Tuple[np.dtype, Any]: # pragma: no co return my_dtype, my_ctype -def is_pickleable(obj: Any) -> bool: # TODO remove # pragma: no cover - try: - with io.BytesIO() as stream: - pickle.dump(obj, stream) - return True - except: - return False - - # Functionality to import modules/objects by name, and call functions by name # -------------------------------------------------------------------------------------