Fix unused import that causes a runtime error, and clean up other unused imports

Author: amorehead

setup.py

@@ -7,7 +7,7 @@
     version="0.0.1",
     description="A hub for deep diffusion networks designed to generate novel biological data",
     author="Alex Morehead",
-    author_email="[email protected]",
+    author_email="[email protected]",
     url="https://github.com/BioinfoMachineLearning/bio-diffusion",
     install_requires=["pytorch-lightning", "hydra-core"],
     packages=find_packages(),

src/datamodules/components/edm/__init__.py

@@ -2,15 +2,13 @@
 From https://raw.githubusercontent.com/arneschneuing/DiffSBDD/
 """
 
-import os
 import torch
-import subprocess
 
 import numpy as np
 
 import src.datamodules.components.edm.constants as edm_constants
 
-from typing import Any, Dict, List, Literal, Tuple, Union
+from typing import Any, Dict, List, Tuple, Union
 
 from torchtyping import TensorType, patch_typeguard
 from typeguard import typechecked

src/datamodules/components/edm/dataset.py

@@ -5,14 +5,10 @@
 import torch
 import os
 
-import torch.nn as nn
 import src.datamodules.components.edm.build_geom_dataset as build_geom_dataset
 
 from omegaconf import DictConfig
 from functools import partial
-from typing import Optional
-
-from src.datamodules.components.edm import download_dataset
 
 from torch.utils.data.dataloader import DataLoader as TorchDataLoader
 from torch_geometric.loader.dataloader import DataLoader as PyGDataLoader