Move inference_api to separate folder

deruyter92 · deruyter92 · commit f61efa5a7550 · 2026-02-13T17:03:41.000+01:00
diff --git a/fmpose3d/__init__.py b/fmpose3d/__init__.py
@@ -29,7 +29,7 @@
 )
 
 # High-level inference API
-from .fmpose3d import (
+from .inference_api.fmpose3d import (
     FMPose3DInference,
     HRNetEstimator,
     Pose2DResult,
diff --git a/fmpose3d/inference_api/__init__.py b/fmpose3d/inference_api/__init__.py
@@ -0,0 +1,17 @@
+"""FMPose3D inference API."""
+
+from .fmpose3d import (
+    FMPose3DInference,
+    HRNetEstimator,
+    Pose2DResult,
+    Pose3DResult,
+    Source,
+)
+
+__all__ = [
+    "FMPose3DInference",
+    "HRNetEstimator",
+    "Pose2DResult",
+    "Pose3DResult",
+    "Source",
+]
diff --git a/fmpose3d/inference_api/fmpose3d.py b/fmpose3d/inference_api/fmpose3d.py
diff --git a/tests/fmpose3d_api/test_fmpose3d.py b/tests/fmpose3d_api/test_fmpose3d.py
@@ -19,7 +19,7 @@
 import pytest
 import torch
 
-from fmpose3d.fmpose3d import (
+from fmpose3d.inference_api.fmpose3d import (
     AnimalPostProcessor,
     FMPose3DInference,
     HRNetEstimator,
diff --git a/tests/fmpose3d_api/test_huggingface.py b/tests/fmpose3d_api/test_huggingface.py
@@ -20,7 +20,7 @@
 
 import pytest
 
-from fmpose3d.fmpose3d import FMPose3DInference
+from fmpose3d.inference_api.fmpose3d import FMPose3DInference
 
 
 class TestDownloadModelWeights:
diff --git a/tests/fmpose3d_api/test_pipeline.py b/tests/fmpose3d_api/test_pipeline.py
@@ -36,7 +36,7 @@
 import numpy as np
 import pytest
 
-from fmpose3d.fmpose3d import FMPose3DInference
+from fmpose3d.inference_api.fmpose3d import FMPose3DInference
 
 from .conftest import (
     requires_human_weights,

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@`
`29`	`29`	`)`
`30`	`30`
`31`	`31`	`# High-level inference API`
`32`		`-from .fmpose3d import (`
	`32`	`+from .inference_api.fmpose3d import (`
`33`	`33`	`FMPose3DInference,`
`34`	`34`	`HRNetEstimator,`
`35`	`35`	`Pose2DResult,`