Skip to content

Commit

Permalink
chore: Refactor SLURM training script for robot_sf
Browse files Browse the repository at this point in the history
The SLURM training script `slurm_PPO_robot_sf.py` has been refactored to improve code readability and maintainability. The unnecessary import statements have been removed, and the script now follows PEP 8 style guidelines. This refactor does not change the functionality of the script, but it enhances the overall quality of the codebase.
  • Loading branch information
ll7 committed Sep 17, 2024
1 parent 41e8e34 commit 0e46529
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions SLURM/slurm_PPO_robot_sf.py
Original file line number Diff line number Diff line change
@@ -1,12 +1,7 @@
"train a robot in robot_sf on a slurm server"
"""train a robot in robot_sf on a slurm server"""

"""
This script is used to train a PPO model on the CartPole-v1 environment.
"""
import sys
from datetime import datetime

import gymnasium as gym
from loguru import logger

from stable_baselines3 import PPO
Expand Down

0 comments on commit 0e46529

Please sign in to comment.