minor changes

This commit is contained in:
Dominik Moritz Roth 2023-07-06 18:20:37 +02:00
parent 361c596d4d
commit 55617fece4
3 changed files with 9 additions and 7 deletions

1
.gitignore vendored
View File

@ -1,3 +1,4 @@
__pycache__
.venv
wandb
*.egg-info/

View File

@ -1 +1 @@
from slate import Slate
from slate.slate import Slate

View File

@ -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...')