Dusan commited on
Commit
6526b0f
1 Parent(s): fb36ee2

Update fudge/data.py

Browse files
Files changed (1) hide show
  1. fudge/data.py +3 -3
fudge/data.py CHANGED
@@ -12,9 +12,9 @@ import numpy as np
12
  from tqdm import tqdm
13
  import torch
14
 
15
- from data.util import suppress_stdout
16
- from data.poetry_util import is_iambic, count_syllables, get_rhymes, get_rhyme_group
17
- from data.constants import *
18
 
19
  DatasetInfo = namedtuple('DatasetInfo',
20
  ['index2word', 'word2index', 'total_words', 'vocab', 'glove_embeddings'])
 
12
  from tqdm import tqdm
13
  import torch
14
 
15
+ from fudge.util import suppress_stdout
16
+ from fudge.poetry_util import is_iambic, count_syllables, get_rhymes, get_rhyme_group
17
+ from fudge.constants import *
18
 
19
  DatasetInfo = namedtuple('DatasetInfo',
20
  ['index2word', 'word2index', 'total_words', 'vocab', 'glove_embeddings'])