From d3afac437b1ddf563c8d2694ada950bf20ab1c34 Mon Sep 17 00:00:00 2001
From: will-thompson-k <19388895+will-thompson-k@users.noreply.github.com>
Date: Wed, 9 Dec 2020 20:08:36 -0600
Subject: [PATCH] ** commented out test. It takes up too much memory on Travis
 CI.

---
 tests/test_trainer.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/test_trainer.py b/tests/test_trainer.py
index 796b829..4e9be51 100644
--- a/tests/test_trainer.py
+++ b/tests/test_trainer.py
@@ -1,9 +1,12 @@
 from argparse import Namespace
 
+import pytest
+
 import torch
 from nlpmodels.models import gpt, text_cnn, transformer, word2vec
 from nlpmodels.utils import utils, train
 from nlpmodels.utils.elt import skipgram_dataset, gpt_dataset, text_cnn_dataset, transformer_dataset
+
 import os
 
 # mac issue
@@ -107,7 +110,7 @@ def test_text_cnn_trainer_regression_test():
     assert losses[0].data >= losses[-1].data
 
 
-# @pytest.mark.skip(reason="upgrading torchtext versions to 0.8.0")
+@pytest.mark.skip(reason="upgrading torchtext versions to 0.8.0")
 def test_transformer_trainer_regression_test():
 
     utils.set_seed_everywhere()