update

app.py

@@ -84,7 +84,10 @@ def regression_transformer(
     )
     model = RegressionTransformer(configuration=config, target=target)
     samples = list(model.sample(number_of_samples))
-    if algorithm_version == "polymer" and task == "Generate":
+    polymer = isinstance(
+        config.generator.tokenizer.text_tokenizer, PolymerGraphTokenizer
+    )
+    if algorithm_version == "rop_catalyst" and task == "Generate":
         correct_samples = [(s, p) for s, p in samples if "." in s]
         while len(correct_samples) < number_of_samples:
             samples = list(model.sample(number_of_samples))
@@ -96,8 +99,7 @@ def regression_transformer(
                 ]
             )
         samples = correct_samples
-    # if isinstance(config.generator.tokenizer.text_tokenizer, PolymerGraphTokenizer):
-    #     pass
+
     if task == "Predict":
         return draw_grid_predict(samples[0], target, domain=algorithm.split(":")[0])
     else: