Commit e92a4f20 authored by Joseph Heenan's avatar Joseph Heenan
Browse files

Fixing hardcoded path

parent 8a49ad8d
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -34,7 +34,7 @@ def load_pcba(featurizer='ECFP', split='random', reload=True):
  elif featurizer == 'Raw':
    featurizer = deepchem.feat.RawFeaturizer()

  with gzip.GzipFile("/media/data/pubchem/pcba.csv.gz", "r") as file:
  with gzip.GzipFile(dataset_file, "r") as file:
    header = file.readline().rstrip().decode("utf-8")
    columns = header.split(",")
    columns.remove("mol_id")
+1 −1

File changed.

Contains only whitespace changes.