CorrDiff: Remove pickle dependency (#445)

* remove pickle

* Remove unused io library

---------

Co-authored-by: Kaustubh Tangsali <71059996+ktangsali@users.noreply.github.com>

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

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,

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
 # -------------------------------------------------------------------------------------