minor changes
This commit is contained in:
parent
361c596d4d
commit
55617fece4
1
.gitignore
vendored
1
.gitignore
vendored
@ -1,3 +1,4 @@
|
||||
__pycache__
|
||||
.venv
|
||||
wandb
|
||||
*.egg-info/
|
||||
|
@ -1 +1 @@
|
||||
from slate import Slate
|
||||
from slate.slate import Slate
|
||||
|
@ -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...')
|
||||
|
Loading…
Reference in New Issue
Block a user