Update graph_decoder/diffusion_model.py

graph_decoder/diffusion_model.py

@@ -7,9 +7,9 @@ import torch.nn as nn
 import torch.nn.functional as F
 
 # from . import diffusion_utils as utils
-# from .molecule_utils import graph_to_smiles, check_valid
+from .molecule_utils import graph_to_smiles, check_valid
 # from .transformer import Transformer
-# from .visualize_utils import MolecularVisualization
+from .visualize_utils import MolecularVisualization
 
 class GraphDiT(nn.Module):
     def __init__(
@@ -27,7 +27,7 @@ class GraphDiT(nn.Module):
         pass
 
     def generate(self, properties, guide_scale, num_nodes, number_chain_steps):
-        pass
+        return 0, 0
     
 
 # class GraphDiT(nn.Module):