Signed-off-by: Jeroen Oudshoorn <oudshoorn.jeroen@gmail.com>
This commit is contained in:
Jeroen Oudshoorn
2023-10-06 23:32:18 +02:00
parent 00cefd06fe
commit 36cd9d8dd9
4 changed files with 6 additions and 5 deletions

View File

@ -1,6 +1,6 @@
import logging
import gymnasium
from gymnasium import spaces
import gym
from gym import spaces
import numpy as np
import pwnagotchi.ai.featurizer as featurizer
@ -8,7 +8,7 @@ import pwnagotchi.ai.reward as reward
from pwnagotchi.ai.parameter import Parameter
class Environment(gymnasium.Env):
class Environment(gym.Env):
metadata = {'render.modes': ['human']}
params = [
Parameter('min_rssi', min_value=-200, max_value=-50),

View File

@ -1,4 +1,4 @@
from gymnasium import spaces
from gym import spaces
class Parameter(object):