diff --git a/dreamer.py b/dreamer.py
index fba6bce..7d41a55 100644
--- a/dreamer.py
+++ b/dreamer.py
@@ -1,10 +1,8 @@
 import argparse
-import collections
 import functools
 import os
 import pathlib
 import sys
-import warnings
 
 os.environ["MUJOCO_GL"] = "egl"
 
diff --git a/tools.py b/tools.py
index 59da93a..f7ee068 100644
--- a/tools.py
+++ b/tools.py
@@ -3,7 +3,6 @@ import collections
 import io
 import json
 import pathlib
-import pickle
 import re
 import time
 import uuid
@@ -14,7 +13,6 @@ import torch
 from torch import nn
 from torch.nn import functional as F
 from torch import distributions as torchd
-from torch.utils.data import Dataset
 from torch.utils.tensorboard import SummaryWriter