fix creation of Body with pose=None (#22)

Tau-J · May 30, 2024 · da0d3c3 · da0d3c3
1 parent f1fce5e
commit da0d3c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rtmlib/tools/solution/body.py b/rtmlib/tools/solution/body.py
@@ -99,7 +99,7 @@ def __init__(self,
                  backend: str = 'onnxruntime',
                  device: str = 'cpu'):
 
-        if 'rtmo' in pose:
+        if pose is not None and 'rtmo' in pose:
             from .. import RTMO
 
             self.one_stage = True