Skip to content

Commit

Permalink
Clean up the import in embedders
Browse files Browse the repository at this point in the history
Remove the repeated imports of torch and unused module Batch and osp
  • Loading branch information
xiaoruiDong committed Mar 14, 2024
1 parent b856259 commit 9dd1869
Showing 1 changed file with 1 addition and 8 deletions.
9 changes: 1 addition & 8 deletions rdmc/conformer_generation/embedders.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,19 +8,12 @@
from pathlib import Path

from rdmc import RDKitMol
import os.path as osp

import numpy as np
from time import time

try:
import torch
from torch_geometric.data import Batch
except ImportError:
pass

from .utils import *

# GeoMol relevant imports
try:
import torch
from geomol.model import GeoMol
Expand Down

0 comments on commit 9dd1869

Please sign in to comment.