diff --git a/src/axolotl/integrations/rrt/__init__.py b/src/axolotl/integrations/rrt/__init__.py index 75681306a..976897ea6 100644 --- a/src/axolotl/integrations/rrt/__init__.py +++ b/src/axolotl/integrations/rrt/__init__.py @@ -20,7 +20,7 @@ class RelaxedRecursiveTransformerPlugin(BasePlugin): """ def get_input_args(self): - return "axolotl.integrations.rrt.RelaxedRecursiveTransformerArgs" + return "axolotl.integrations.rrt.args.RelaxedRecursiveTransformerArgs" def register(self): LOG.info(