【Fix】 fix cell pose

STOmics · Nov 19, 2023 · d2ec44b · d2ec44b
1 parent 02ca368
commit d2ec44b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/stereo/algorithm/cell_pose/__init__.py b/stereo/algorithm/cell_pose/__init__.py
@@ -2,9 +2,9 @@
 # coding: utf-8
 # flake8: noqa
 try:
-    from stereo.algorithm.cell_pose.cell_pose import Cellpose
+    from stereo.algorithm.cell_pose.cell_pose import CellPose as Cellpose
 except ImportError:
-    errmsg = """class `Cellpose` is not import.
+    errmsg = """class `CellPose` is not import.
 ************************************************
 * Some necessary modules may not be installed. *
 * Please install them by:                      *

diff --git a/stereo/algorithm/cell_pose/cell_pose.py b/stereo/algorithm/cell_pose/cell_pose.py
@@ -3,7 +3,7 @@
 """
 @author: zhen bin  [email protected]
 @last modified by: zhen bin
-@file:common.py
+@file:cell_pose.py
 @time:2023/08/24
 """
 from math import ceil
@@ -19,7 +19,7 @@
 from stereo.utils.time_consume import log_consumed_time
 
 
-class Cellpose:
+class CellPose:
 
     def __init__(self,
                  img_path: str,