Remove dead code across codebase

pufferlib/emulation.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import numpy as np
 import warnings

pufferlib/environments/__init__.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import pufferlib
 
 def try_import(module_path, package_name=None):

pufferlib/environments/atari/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import functools
 

pufferlib/environments/box2d/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gymnasium
 import functools

pufferlib/environments/bsuite/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import gym
 import functools
 

pufferlib/environments/butterfly/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 from pettingzoo.utils.conversions import aec_to_parallel_wrapper
 import functools
 

pufferlib/environments/crafter/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gym
 import gymnasium

pufferlib/environments/dm_control/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gym
 import shimmy

pufferlib/environments/dm_lab/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gym
 import shimmy

pufferlib/environments/griddly/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gym
 import shimmy

pufferlib/environments/gvgai/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import functools
 

pufferlib/environments/links_awaken/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gymnasium
 

pufferlib/environments/magent/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 from pettingzoo.utils.conversions import aec_to_parallel_wrapper
 import functools
 

pufferlib/environments/microrts/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import warnings

pufferlib/environments/minerl/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gym
 import shimmy

pufferlib/environments/minigrid/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gymnasium
 import functools

pufferlib/environments/minihack/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gym
 import shimmy

pufferlib/environments/minihack/torch.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import pufferlib.pytorch
 from pufferlib.environments.nethack import Policy

pufferlib/environments/mujoco/environment.py

@@ -1,5 +1,4 @@
 
-from pdb import set_trace as T
 
 import functools
 

pufferlib/environments/nethack/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import shimmy
 import gym

pufferlib/environments/nethack/torch.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import torch
 import torch.nn as nn

pufferlib/environments/nmmo/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import functools
 

pufferlib/environments/nmmo/torch.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import torch
 import torch.nn.functional as F

pufferlib/environments/open_spiel/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import functools
 

pufferlib/environments/open_spiel/gymnasium_environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 from open_spiel.python.algorithms import mcts

pufferlib/environments/open_spiel/pettingzoo_environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import pufferlib

pufferlib/environments/open_spiel/torch.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import torch

pufferlib/environments/open_spiel/utils.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import gymnasium

pufferlib/environments/pokemon_red/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 
 import gymnasium
 import functools

pufferlib/environments/procgen/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import gym

pufferlib/environments/procgen/torch.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 from torch import nn
 import pufferlib.models
 

pufferlib/environments/slimevolley/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import functools
 

pufferlib/environments/stable_retro/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import gymnasium as gym

pufferlib/environments/test/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import time

pufferlib/environments/test/mock_environments.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import time

pufferlib/environments/vizdoom/environment.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import functools
 

pufferlib/models.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 
 import torch

pufferlib/ocean/environment.py

@@ -1,14 +1,6 @@
 import importlib
 import pufferlib.emulation
 
-def lazy_import(module_path, attr):
-    """
-    Returns a callable that, when called with any arguments, will
-    import the module, retrieve the attribute (usually a class or factory)
-    and then call it with the given arguments.
-    """
-    return lambda *args, **kwargs: getattr(__import__(module_path, fromlist=[attr]), attr)(*args, **kwargs)
-
 def make_foraging(width=1080, height=720, num_agents=4096, horizon=512,
         discretize=True, food_reward=0.1, render_mode='rgb_array'):
     from .grid import grid

pufferlib/ocean/moba/moba.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 import os
 

pufferlib/ocean/nmmo3/nmmo3.py

@@ -1,4 +1,3 @@
-from pdb import set_trace as T
 import numpy as np
 from types import SimpleNamespace
 import gymnasium

pufferlib/pufferl.py

@@ -713,12 +713,6 @@ def dist_sum(value, device):
     torch.distributed.all_reduce(tensor, op=torch.distributed.ReduceOp.SUM)
     return tensor.item()
 
-def dist_mean(value, device):
-    if not torch.distributed.is_initialized():
-        return value
-
-    return dist_sum(value, device) / torch.distributed.get_world_size()
-
 class Profile:
     def __init__(self, frequency=5):
         self.profiles = defaultdict(lambda: defaultdict(float))

pufferlib/pytorch.py

@@ -1,11 +1,9 @@
 import sys
-from pdb import set_trace as T
 from typing import Dict, List, Tuple, Union
 
 import numpy as np
 import torch
 from torch import nn
-from torch.distributions import Categorical
 from torch.distributions.utils import logits_to_probs
 
 import pufferlib
@@ -103,23 +101,6 @@ def nativize_tensor(observation: torch.Tensor, native_dtype: NativeDType):
     return _nativize_tensor(observation, native_dtype)
 
 
-# torch.view(dtype) does not compile
-# This is a workaround hack
-# @thatguy - can you figure out a more robust way to handle cast?
-# I think it may screw up for non-uint data... so I put a hard .view
-# fallback that breaks compile
-def compilable_cast(u8, dtype):
-    if dtype in (torch.uint8, torch.uint16, torch.uint32, torch.uint64):
-        n = dtype.itemsize
-        bytes = [u8[..., i::n].to(dtype) for i in range(n)]
-        if not LITTLE_BYTE_ORDER:
-            bytes = bytes[::-1]
-
-        bytes = sum(bytes[i] << (i * 8) for i in range(n))
-        return bytes.view(dtype)
-    return u8.view(dtype)  # breaking cast
-
-
 def _nativize_tensor(observation: torch.Tensor, native_dtype: NativeDType):
     if isinstance(native_dtype, tuple):
         dtype, shape, offset, delta = native_dtype
@@ -141,15 +122,6 @@ def _nativize_tensor(observation: torch.Tensor, native_dtype: NativeDType):
         return subviews
 
 
-def nativize_observation(observation, emulated):
-    # TODO: Any way to check that user has not accidentally cast data to float?
-    # float is natively supported, but only if that is the actual correct type
-    return nativize_tensor(
-        observation,
-        emulated['observation_dtype'],
-        emulated['emulated_observation_dtype'],
-    )
-
 def flattened_tensor_size(native_dtype):
     return _flattened_tensor_size(native_dtype)
 
@@ -182,10 +154,6 @@ def entropy(logits):
     p_log_p = logits * logits_to_probs(logits)
     return -p_log_p.sum(-1)
 
-def entropy_probs(logits, probs):
-    p_log_p = logits * probs
-    return -p_log_p.sum(-1)
-
 def sample_logits(logits, action=None):
     is_discrete = isinstance(logits, torch.Tensor)
     if isinstance(logits, torch.distributions.Normal):

pufferlib/vector.py

@@ -1,6 +1,5 @@
 # TODO: Check actions passed to envs are right shape? On first call at least
 
-from pdb import set_trace as T
 
 import numpy as np
 import time