Skip to content

GitLab

Unverified Commit 00f33224 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

[GCMC] Fixing a parser warning appearing in one of the `gcmc` examples. (#7118)

parent b0080d5b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
examples/pytorch/gcmc/data.py
View file @ 00f33224
...@@ -689,6 +689,7 @@ class MovieLens(object): ...@@ -689,6 +689,7 @@ class MovieLens(object):
header=None, header=None,
names=["id", "title", "genres"], names=["id", "title", "genres"],
encoding="iso-8859-1", encoding="iso-8859-1",
engine="python",
) )
genre_map = {ele: i for i, ele in enumerate(GENRES)} genre_map = {ele: i for i, ele in enumerate(GENRES)}
genre_map["Children's"] = genre_map["Children"] genre_map["Children's"] = genre_map["Children"]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment