Revert "disable pytorch (#386)"

This reverts commit 311a6f4.

Author: NihalHarish

tests/zero_code_change/test_pytorch_integration.py

@@ -12,7 +12,6 @@
 
 # Third Party
 import pytest
-import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.optim as optim
@@ -23,10 +22,6 @@
 from smdebug.core.utils import SagemakerSimulator, ScriptSimulator
 
 
-@pytest.mark.skipif(
-    torch.__version__ == "1.7.0",
-    reason="Disabling the test temporarily until we root cause the version incompatibility",
-)
 @pytest.mark.parametrize("script_mode", [False])
 @pytest.mark.parametrize("use_loss_module", [True, False])
 def test_pytorch(script_mode, use_loss_module):