Fix the case of use decorator directly to raw class and add test case (#4177)

2026-06-28 04:23:03 +08:00 · 2019-02-28 00:09:42 +08:00
parent db5c3b22b7
commit 0a11b27971
2 changed files with 14 additions and 5 deletions
@@ -664,7 +664,11 @@ class FunctionActorManager(object):
                                   function_descriptor.class_name)
        try:
            module = importlib.import_module(module_name)
-            return getattr(module, class_name)._modified_class
+            actor_class = getattr(module, class_name)
+            if isinstance(actor_class, ray.actor.ActorClass):
+                return actor_class._modified_class
+            else:
+                return actor_class
        except Exception:
            logger.exception(
                "Failed to load actor_class %s.".format(class_name))