Spaces:
Sleeping
Sleeping
use smaller
Browse files- rlcube/main.py +1 -1
rlcube/main.py
CHANGED
|
@@ -25,7 +25,7 @@ def solve(body: StateArgs):
|
|
| 25 |
|
| 26 |
env = Cube2Env()
|
| 27 |
env.reset(state=np.array(state, dtype=np.int8))
|
| 28 |
-
tree = MonteCarloTree(env.obs(), max_simulations=
|
| 29 |
if tree.is_solved:
|
| 30 |
return {"steps": [action for _, action in tree.solved_path]}
|
| 31 |
raise HTTPException(status_code=422, detail="Unable to solve the cube")
|
|
|
|
| 25 |
|
| 26 |
env = Cube2Env()
|
| 27 |
env.reset(state=np.array(state, dtype=np.int8))
|
| 28 |
+
tree = MonteCarloTree(env.obs(), max_simulations=256)
|
| 29 |
if tree.is_solved:
|
| 30 |
return {"steps": [action for _, action in tree.solved_path]}
|
| 31 |
raise HTTPException(status_code=422, detail="Unable to solve the cube")
|