Update inference.py

inference.py

@@ -89,8 +89,11 @@ class Inference(object):
                                       bond_encoder=self.bond_encoder,
                                       bond_decoder=self.bond_decoder)
 
-        self.sample_num = config.sample_num if config.sample_num < self.inf_dataset.num_smiles else self.inf_dataset.num_smiles
-    
+        try:
+            self.sample_num = config.sample_num if config.sample_num < self.inf_dataset.num_smiles else self.inf_dataset.num_smiles
+        except:
+            self.sample_num = config.sample_num
+
         self.inf_loader = DataLoader(self.inf_dataset,
                                  shuffle=True,
                                  batch_size=self.inf_batch_size,