commit 1af7df6e78b8c000e4ec1321a3d928bdd6323578
author Alexander Sidorov <asidorov566@gmail.com> Thu Mar 01 21:02:52 2018 -0800
committer GitHub <noreply@github.com> Thu Mar 01 21:02:52 2018 -0800
tree cf5ee3998594e7d800719ceca100bb29ab5f1629
parent acd8dfdfb94e96ce112d3f3a1c8efb7dc9f563fe

fix rnn_cell_test in fbcode (#2107)

caffe2/python/operator_test/rnn_cell_test.py

diff --git a/caffe2/python/operator_test/rnn_cell_test.py b/caffe2/python/operator_test/rnn_cell_test.py
index a23b891..cfdb978 100644
--- a/caffe2/python/operator_test/rnn_cell_test.py
+++ b/caffe2/python/operator_test/rnn_cell_test.py
@@ -32,7 +32,6 @@
 from hypothesis import settings as ht_settings
 import hypothesis.strategies as st
 import numpy as np
-import pytest
 import unittest
 
 
@@ -1097,13 +1096,13 @@
     def test_milstm(self):
         self.lstm_base(lstm_type=(rnn_cell.MILSTM, milstm_reference))
 
-    @pytest.mark.skip(reason="This is currently numerically unstable")
+    @unittest.skip("This is currently numerically unstable")
     def test_norm_lstm(self):
         self.lstm_base(
             lstm_type=(rnn_cell.LayerNormLSTM, layer_norm_lstm_reference),
         )
 
-    @pytest.mark.skip(reason="This is currently numerically unstable")
+    @unittest.skip("This is currently numerically unstable")
     def test_norm_milstm(self):
         self.lstm_base(
             lstm_type=(rnn_cell.LayerNormMILSTM, layer_norm_milstm_reference)