From 55617fece47f6717b1a531fedcd675c3c99e41bf Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Thu, 6 Jul 2023 18:20:37 +0200
Subject: [PATCH] minor changes

---
 .gitignore        |  1 +
 slate/__init__.py |  2 +-
 slate/slate.py    | 13 +++++++------
 3 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/.gitignore b/.gitignore
index d31b356..6e18e7b 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,4 @@
 __pycache__
 .venv
 wandb
+*.egg-info/
diff --git a/slate/__init__.py b/slate/__init__.py
index a149628..86b99bd 100644
--- a/slate/__init__.py
+++ b/slate/__init__.py
@@ -1 +1 @@
-from slate import Slate
+from slate.slate import Slate
diff --git a/slate/slate.py b/slate/slate.py
index 2e9b5e3..0cd7ea7 100644
--- a/slate/slate.py
+++ b/slate/slate.py
@@ -1,10 +1,4 @@
-#import fancy_gym
-#from stable_baselines3 import PPO
-#from stable_baselines3.common.monitor import Monitor
-#from stable_baselines3.common.vec_env import DummyVecEnv, VecVideoRecorder
 import wandb
-from wandb.integration.sb3 import WandbCallback
-#import gymnasium as gym
 import yaml
 import os
 import random
@@ -32,6 +26,7 @@ else:
 class Slate():
     def __init__(self, runners):
         self.runners = runners
+        self.runners['printConfig'] = print_config_runner
 
     def load_config(self, filename, name):
         config, stack = self._load_config(filename, name)
@@ -202,5 +197,11 @@ class Slate():
             self.run_local(args.config_file, args.experiment, args.job_num)
 
 
+def print_config_runner(slate, run, config):
+    print(config)
+    for k in list(config.keys()):
+        del config[k]
+
+
 if __name__ == '__main__':
     raise Exception('You are using it wrong...')