undo skip for gh actions

tlpss · Oct 25, 2023 · a39cd64 · a39cd64
1 parent 0acca23
commit a39cd64
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/test/test_detector.py b/test/test_detector.py
@@ -1,7 +1,6 @@
 import os
 import unittest
 
-import pytest
 import torch
 from pytorch_lightning.loggers import WandbLogger
 from torch import nn
@@ -104,7 +103,7 @@ def test_model_init_heatmaps(self):
     # TODO: chcek if we can run it on gh actions as well.
     IN_GITHUB_ACTIONS = os.getenv("GITHUB_ACTIONS") == "true"
 
-    @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Test doesn't work in Github Actions atm")
+    # @pytest.mark.skipif(IN_GITHUB_ACTIONS, reason="Test doesn't work in Github Actions atm")
     def test_checkpoint_loading(self):
         wandb_logger = WandbLogger(dir=get_wandb_log_dir_path(), mode="offline")