Skip to content
This repository has been archived by the owner on Jul 12, 2021. It is now read-only.

Commit

Permalink
** commented out test. It takes up too much memory on Travis CI.
Browse files Browse the repository at this point in the history
  • Loading branch information
will-thompson-k committed Dec 10, 2020
1 parent ae34182 commit d3afac4
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion tests/test_trainer.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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()
Expand Down

0 comments on commit d3afac4

Please sign in to comment.