fix tpe exception (#209)

aistairc · Feb 8, 2023 · b5d2a40 · b5d2a40
1 parent e5a6ddb
commit b5d2a40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aiaccel/optimizer/tpe_optimizer.py b/aiaccel/optimizer/tpe_optimizer.py
@@ -226,6 +226,6 @@ def create_distributions(
             distributions[p.name] = optuna.distributions.CategoricalDistribution(p.sequence)
 
         else:
-            raise 'Unsupported parameter type'
+            raise TypeError('Unsupported parameter type')
 
     return distributions