Fix the definitions of felis.free_dict.identity, `felis.free_t.iden…

…tity` and `felis.free.identity`

src/felis/free.py

@@ -2,10 +2,10 @@
 
 from collections.abc import Callable
 from dataclasses import dataclass
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 import felis.identity
-from felis import applicative, free_t, function, identity, monad
+from felis import applicative, free_t, function, monad
 from felis.currying import curry
 from felis.free_t import Pure
 
@@ -54,6 +54,13 @@ def map[From, To](free_value: Free[From], function: Callable[[From], To]) -> Fre
     map = free_t.map(felis.identity.map)
 
 
+if TYPE_CHECKING:
+    # [T : Type] -> T -> Free T
+    identity: Free[Any]
+else:
+    from felis.free_t import identity
+
+
 if TYPE_CHECKING:
 
     @curry

src/felis/free_dict.py

@@ -2,11 +2,11 @@
 
 from collections.abc import Callable
 from dataclasses import dataclass
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 
 from felis import applicative, dict, free_t, function, monad
 from felis.currying import curry
-from felis.free_t import Pure, identity
+from felis.free_t import Pure
 
 __all__ = [
     "Pure",
@@ -53,6 +53,13 @@ def map[K, From, To](free_dict_value: FreeDict[K, From], function: Callable[[Fro
     map = free_t.map(dict.map)
 
 
+if TYPE_CHECKING:
+    # [K : Type] -> [T : Type] -> T -> FreeDict K T
+    identity: FreeDict[Any, Any]
+else:
+    from felis.free_t import identity
+
+
 if TYPE_CHECKING:
 
     @curry

src/felis/free_t.py

@@ -1,6 +1,6 @@
 from collections.abc import Callable
 from dataclasses import dataclass
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 import felis.function
 from felis.currying import curry
@@ -35,7 +35,11 @@ def map[From, To](free_value: Free[From], function: Callable[[From], To], f_map:
             return Bind(f_map(map(f_map)(function))(f_free_value))
 
 
-identity = Pure
+if TYPE_CHECKING:
+    # [F : Type -> Type] -> [T : Type] -> T -> Free F T
+    identity: Free[Any]
+else:
+    identity = Pure
 
 
 # [F : Type -> Type] ->