Fix pre-commit issues

alex-petrenko · Dec 27, 2023 · 6379cf9 · 6379cf9
1 parent de799b5
commit 6379cf9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -1,14 +1,14 @@
 repos:
 -   repo: https://github.com/psf/black
-    rev: 23.1.0
+    rev: 23.12.1
     hooks:
     -   id: black
 -   repo: https://github.com/pycqa/isort
-    rev: 5.12.0
+    rev: 5.13.2
     hooks:
       - id: isort
         name: isort (python)
 -   repo: https://github.com/pycqa/flake8
-    rev: 6.0.0
+    rev: 6.1.0
     hooks:
     -   id: flake8
diff --git a/sample_factory/model/actor_critic.py b/sample_factory/model/actor_critic.py
@@ -76,7 +76,7 @@ def initialize_weights(self, layer):
             layer.bias.data.fill_(0)
 
         if self.cfg.policy_initialization == "orthogonal":
-            if type(layer) == nn.Conv2d or type(layer) == nn.Linear:
+            if type(layer) is nn.Conv2d or type(layer) is nn.Linear:
                 nn.init.orthogonal_(layer.weight.data, gain=gain)
             else:
                 # LSTMs and GRUs initialize themselves
@@ -85,7 +85,7 @@ def initialize_weights(self, layer):
                 # go with default initialization,
                 pass
         elif self.cfg.policy_initialization == "xavier_uniform":
-            if type(layer) == nn.Conv2d or type(layer) == nn.Linear:
+            if type(layer) is nn.Conv2d or type(layer) is nn.Linear:
                 nn.init.xavier_uniform_(layer.weight.data, gain=gain)
             else:
                 pass