Skip to content

Commit 68fb0b7

Browse files
committed
fix flake8 issues
1 parent 6ee3115 commit 68fb0b7

File tree

3 files changed

+3
-14
lines changed

3 files changed

+3
-14
lines changed

examples/rnngen/infer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from model.model import CharRNN
1111
from model.vocab import START_CHAR, END_CHAR
12-
from train import getconfig, get_vocab_from_file
12+
from train import get_vocab_from_file
1313

1414

1515
def count_valid_samples(smiles, rdkit=True):
@@ -79,7 +79,7 @@ def sample(model, i2c, c2i, device, temp=1, batch_size=10, max_len=150):
7979

8080

8181
def main(args, device):
82-
config = getconfig(args)
82+
# config = getconfig(args)
8383
print("loading data.")
8484
vocab, c2i, i2c, _, _ = get_vocab_from_file(args.i + "/vocab.txt")
8585

examples/rnngen/model/vocab.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,7 @@ def main(args):
9090
_, c2i, _, _, _ = get_vocab_from_file(args.o + '/vocab.txt')
9191

9292
# seconnd step is to make data:
93-
count_ = count
93+
# count_ = count
9494
count = 0
9595

9696
with open(args.i, 'r') as f:

examples/rnngen/train.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,3 @@
1-
import argparse
2-
from model.vocab import get_vocab_from_file, START_CHAR, END_CHAR
3-
from model.model import CharRNN
4-
import torch.utils.data
5-
import numpy as np
6-
import torch
7-
import torch.nn as nn
8-
import torch.nn.utils.rnn
9-
import torch.nn.functional as F
10-
from tqdm import tqdm
11-
import os
121

132

143
def getconfig(args):

0 commit comments

Comments
 (0)