add black profile

Farama-Foundation · Jan 26, 2022 · a75b305 · a75b305
1 parent e794ac9
commit a75b305
Show file tree

Hide file tree

Showing 5 changed files with 33 additions and 21 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -4,7 +4,8 @@ repos:
     hooks:
       - id: isort
         args:
-          - --skip wandb
+          - --profile=black
+          - --skip=wandb
   - repo: /~https://github.com/myint/autoflake
     rev: v1.4
     hooks:

diff --git a/experiments/league.py b/experiments/league.py
@@ -12,9 +12,17 @@
 import numpy as np
 import pandas as pd
 import torch
-from peewee import (JOIN, CharField, DateTimeField, FloatField,
-                    ForeignKeyField, Model, SmallIntegerField, SqliteDatabase,
-                    fn)
+from peewee import (
+    JOIN,
+    CharField,
+    DateTimeField,
+    FloatField,
+    ForeignKeyField,
+    Model,
+    SmallIntegerField,
+    SqliteDatabase,
+    fn,
+)
 from ppo_gridnet import Agent, MicroRTSStatsRecorder
 from stable_baselines3.common.vec_env import VecMonitor
 from trueskill import Rating, quality_1vs1, rate_1vs1

diff --git a/experiments/ppo_gridnet.py b/experiments/ppo_gridnet.py
@@ -13,8 +13,7 @@
 import torch.nn as nn
 import torch.optim as optim
 from gym.spaces import MultiDiscrete
-from stable_baselines3.common.vec_env import (VecEnvWrapper, VecMonitor,
-                                              VecVideoRecorder)
+from stable_baselines3.common.vec_env import VecEnvWrapper, VecMonitor, VecVideoRecorder
 from torch.distributions.categorical import Categorical
 from torch.utils.tensorboard import SummaryWriter
 

diff --git a/gym_microrts/envs/vec_env.py b/gym_microrts/envs/vec_env.py
@@ -82,13 +82,15 @@ def __init__(
         from rts.units import UnitTypeTable
 
         self.real_utt = UnitTypeTable()
-        from ai.rewardfunction import (AttackRewardFunction,
-                                       ProduceBuildingRewardFunction,
-                                       ProduceCombatUnitRewardFunction,
-                                       ProduceWorkerRewardFunction,
-                                       ResourceGatherRewardFunction,
-                                       RewardFunctionInterface,
-                                       WinLossRewardFunction)
+        from ai.rewardfunction import (
+            AttackRewardFunction,
+            ProduceBuildingRewardFunction,
+            ProduceCombatUnitRewardFunction,
+            ProduceWorkerRewardFunction,
+            ResourceGatherRewardFunction,
+            RewardFunctionInterface,
+            WinLossRewardFunction,
+        )
 
         self.rfs = JArray(RewardFunctionInterface)(
             [
@@ -272,13 +274,15 @@ def __init__(
         from rts.units import UnitTypeTable
 
         self.real_utt = UnitTypeTable()
-        from ai.rewardfunction import (AttackRewardFunction,
-                                       ProduceBuildingRewardFunction,
-                                       ProduceCombatUnitRewardFunction,
-                                       ProduceWorkerRewardFunction,
-                                       ResourceGatherRewardFunction,
-                                       RewardFunctionInterface,
-                                       WinLossRewardFunction)
+        from ai.rewardfunction import (
+            AttackRewardFunction,
+            ProduceBuildingRewardFunction,
+            ProduceCombatUnitRewardFunction,
+            ProduceWorkerRewardFunction,
+            ResourceGatherRewardFunction,
+            RewardFunctionInterface,
+            WinLossRewardFunction,
+        )
 
         self.rfs = JArray(RewardFunctionInterface)(
             [

diff --git a/pyproject.toml b/pyproject.toml
@@ -42,4 +42,4 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.extras]
 spyder = ["spyder"]
-cleanrl = ["cleanrl"]
+cleanrl = ["cleanrl"]