Skip to content

Commit 4ce8b87

Browse files
jxcodetwsoumith
authored andcommitted
fix pytorch#229 makes it work on Windows (pytorch#232)
1 parent cddabb2 commit 4ce8b87

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

intermediate_source/char_rnn_classification_tutorial.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767
from __future__ import unicode_literals, print_function, division
6868
from io import open
6969
import glob
70+
import os
7071

7172
def findFiles(path): return glob.glob(path)
7273

@@ -98,7 +99,7 @@ def readLines(filename):
9899
return [unicodeToAscii(line) for line in lines]
99100

100101
for filename in findFiles('data/names/*.txt'):
101-
category = filename.split('/')[-1].split('.')[0]
102+
category = os.path.splitext(os.path.basename(filename))[0]
102103
all_categories.append(category)
103104
lines = readLines(filename)
104105
category_lines[category] = lines

intermediate_source/char_rnn_generation_tutorial.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -77,6 +77,7 @@
7777
from __future__ import unicode_literals, print_function, division
7878
from io import open
7979
import glob
80+
import os
8081
import unicodedata
8182
import string
8283

@@ -102,7 +103,7 @@ def readLines(filename):
102103
category_lines = {}
103104
all_categories = []
104105
for filename in findFiles('data/names/*.txt'):
105-
category = filename.split('/')[-1].split('.')[0]
106+
category = os.path.splitext(os.path.basename(filename))[0]
106107
all_categories.append(category)
107108
lines = readLines(filename)
108109
category_lines[category] = lines

0 commit comments

Comments
 (0)