dodox/Slate
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

minor changes

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

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

2
slate/__init__.py
View File

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

13
slate/slate.py
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 import wandb
from wandb.integration.sb3 import WandbCallback
#import gymnasium as gym
import yaml import yaml
import os import os
import random import random
@ -32,6 +26,7 @@ else:
class Slate(): class Slate():
def __init__(self, runners): def __init__(self, runners):
self.runners = runners self.runners = runners
self.runners['printConfig'] = print_config_runner
def load_config(self, filename, name): def load_config(self, filename, name):
config, stack = self._load_config(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) 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__': if __name__ == '__main__':
raise Exception('You are using it wrong...') raise Exception('You are using it wrong...')