refactor: remove redudant features (#41)

UtilsRL/__init__.py

@@ -1,15 +1,8 @@
 
 __version__ = "0.5.9"
 
-from UtilsRL import data_structure, env, exp, logger, math, misc, plot, rl, monitor
+from UtilsRL import exp, logger
 __all__ = [
-    "data_structure", 
-    "env", 
     "exp", 
     "logger", 
-    "math", 
-    "misc", 
-    "plot", 
-    "rl", 
-    "monitor"
 ]

UtilsRL/exp/__init__.py

@@ -1,8 +1,6 @@
 from UtilsRL.exp.argparse import parse_args, argparse_callbacks, register_argparse_callback
 from UtilsRL.exp._seed import *
 from UtilsRL.exp._device import *
-from UtilsRL.exp.snapshot import make_snapshot
-from UtilsRL.exp.precision import set_precision
 
 from UtilsRL.logger import BaseLogger
 from UtilsRL.logger import logger as url_internal_logger
@@ -63,6 +61,4 @@ def setup(args,
 
     return args
 
-
-register_argparse_callback("UtilsRL.snapshot", make_snapshot)
-register_argparse_callback("UtilsRL.precision", set_precision)
+

UtilsRL/exp/argparse.py

@@ -110,20 +110,6 @@ def traverse_add(old, new, current_key=""):
 
     traverse_add(file_args, cmd_args)
 
-    # check if there is a callback
-    for key in argparse_callbacks:
-        _args = file_args
-        _keys = key.split(".")
-        for k in _keys:
-            _args = _args.get(k, None)
-            if _args is None:
-                ret = argparse_callbacks[key](None)  # call callback with default None and then break
-                file_args = update_args(file_args, ret, eval=False)
-                break
-        else:
-            ret = argparse_callbacks[key](_args)
-            file_args = update_args(file_args, ret, eval=False)
-
     return file_args
 
 def update_args(args, new_args: Optional[Union[dict, list]] = None, eval=True):

UtilsRL/logger/base_logger.py

@@ -1,6 +1,7 @@
 from typing import Any, Dict, Optional, Sequence, Union
 
 import os
+import torch
 import pickle
 from datetime import datetime
 
@@ -26,18 +27,14 @@ def numpy_save(obj, file):
         with open(file, "w") as fp:
             np.save(fp, obj)
 
-    if protocol == "torch":
-        import torch
-        return torch.save
-    else:
-        return {
-            "pickle": pickle_save,
-            "numpy": numpy_save 
-        }.get(protocol)
+    return {
+        "torch": torch.save, 
+        "pickle": pickle_save,
+        "numpy": numpy_save 
+    }.get(protocol)
 
 def load_fn(protocol: str="torch"):
     def torch_load(file):
-        import torch
         return torch.load(file, map_location="cpu")
 
     def pickle_load(file):