clean up nb

Author: RensDimmendaal

fasttransform/transform.py

@@ -52,7 +52,7 @@ def __setitem__(self, k, v):
         if k not in self: super().__setitem__(k, Function(v))
         self[k].dispatch(v)
 
-# %% ../nbs/01_transform.ipynb 14
+# %% ../nbs/01_transform.ipynb 13
 class _TfmMeta(type):
     @classmethod
     def __prepare__(cls, name, bases): return _TfmDict()
@@ -82,7 +82,7 @@ def __new__(cls, name, bases, namespace):
         new_cls.__signature__ = inspect.signature(new_cls.__init__)
         return new_cls
 
-# %% ../nbs/01_transform.ipynb 15
+# %% ../nbs/01_transform.ipynb 14
 class Transform(metaclass=_TfmMeta):
     "Delegates (`__call__`,`decode`,`setup`) to (<code>encodes</code>,<code>decodes</code>,<code>setups</code>) if `split_idx` matches"
     split_idx,init_enc,order,train_setup = None,None,0,None
@@ -137,21 +137,21 @@ def _do_call(self, nm, *args, **kwargs):
 
 add_docs(Transform, decode="Delegate to decodes to undo transform", setup="Delegate to setups to set up transform")
 
-# %% ../nbs/01_transform.ipynb 157
+# %% ../nbs/01_transform.ipynb 156
 class InplaceTransform(Transform):
     "A `Transform` that modifies in-place and just returns whatever it's passed"
     def _call(self, fn, *args, split_idx=None, **kwargs):
         super()._call(fn,*args, split_idx=split_idx, **kwargs)
         return args[0]
 
-# %% ../nbs/01_transform.ipynb 161
+# %% ../nbs/01_transform.ipynb 160
 class DisplayedTransform(Transform):
     "A transform with a `__repr__` that shows its attrs"
 
     @property
     def name(self): return f"{super().name} -- {getattr(self,'__stored_args__',{})}\n"
 
-# %% ../nbs/01_transform.ipynb 167
+# %% ../nbs/01_transform.ipynb 166
 class ItemTransform(Transform):
     "A transform that always take tuples as items"
     _retain = True
@@ -165,21 +165,21 @@ def _call1(self, x, name, **kwargs):
         return retain_type(y, x, Any)
 
 
-# %% ../nbs/01_transform.ipynb 176
+# %% ../nbs/01_transform.ipynb 175
 def get_func(t, name, *args, **kwargs):
     "Get the `t.name` (potentially partial-ized with `args` and `kwargs`) or `noop` if not defined"
     f = nested_callable(t, name)
     return f if not (args or kwargs) else partial(f, *args, **kwargs)
 
-# %% ../nbs/01_transform.ipynb 180
+# %% ../nbs/01_transform.ipynb 179
 class Func():
     "Basic wrapper around a `name` with `args` and `kwargs` to call on a given type"
     def __init__(self, name, *args, **kwargs): self.name,self.args,self.kwargs = name,args,kwargs
     def __repr__(self): return f'sig: {self.name}({self.args}, {self.kwargs})'
     def _get(self, t): return get_func(t, self.name, *self.args, **self.kwargs)
     def __call__(self,t): return mapped(self._get, t)
 
-# %% ../nbs/01_transform.ipynb 183
+# %% ../nbs/01_transform.ipynb 182
 class _Sig():
     def __getattr__(self,k):
         def _inner(*args, **kwargs): return Func(k, *args, **kwargs)
@@ -188,7 +188,7 @@ def _inner(*args, **kwargs): return Func(k, *args, **kwargs)
 Sig = _Sig()
 
 
-# %% ../nbs/01_transform.ipynb 189
+# %% ../nbs/01_transform.ipynb 188
 def compose_tfms(x, tfms, is_enc=True, reverse=False, **kwargs):
     "Apply all `func_nm` attribute of `tfms` on `x`, maybe in `reverse` order"
     if reverse: tfms = reversed(tfms)
@@ -198,13 +198,13 @@ def compose_tfms(x, tfms, is_enc=True, reverse=False, **kwargs):
     return x
 
 
-# %% ../nbs/01_transform.ipynb 194
+# %% ../nbs/01_transform.ipynb 193
 def mk_transform(f):
     "Convert function `f` to `Transform` if it isn't already one"
     f = instantiate(f)
     return f if isinstance(f,(Transform,Pipeline)) else Transform(f)
 
-# %% ../nbs/01_transform.ipynb 195
+# %% ../nbs/01_transform.ipynb 194
 def gather_attrs(o, k, nm):
     "Used in __getattr__ to collect all attrs `k` from `self.{nm}`"
     if k.startswith('_') or k==nm: raise AttributeError(k)
@@ -213,12 +213,12 @@ def gather_attrs(o, k, nm):
     if not res: raise AttributeError(k)
     return res[0] if len(res)==1 else L(res)
 
-# %% ../nbs/01_transform.ipynb 196
+# %% ../nbs/01_transform.ipynb 195
 def gather_attr_names(o, nm):
     "Used in __dir__ to collect all attrs `k` from `self.{nm}`"
     return L(getattr(o,nm)).map(dir).concat().unique()
 
-# %% ../nbs/01_transform.ipynb 197
+# %% ../nbs/01_transform.ipynb 196
 class Pipeline:
     "A pipeline of composed (for encode/decode) transforms, setup with types"
     def __init__(self, funcs=None, split_idx=None):

nbs/01_transform.ipynb

@@ -179,40 +179,6 @@
     "        self[k].dispatch(v)"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "1769fc07",
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "class _TfmMeta(type):\n",
-    "    def __new__(cls, name, bases, dict):\n",
-    "        res = super().__new__(cls, name, bases, dict)\n",
-    "        for nm in _tfm_methods:\n",
-    "            base_td = [getattr(b,nm,None) for b in bases]\n",
-    "            if nm in res.__dict__: getattr(res,nm).bases = base_td\n",
-    "            else: setattr(res, nm, TypeDispatch(bases=base_td))\n",
-    "        # _TfmMeta.__call__ shadows the signature of inheriting classes, set it back\n",
-    "        res.__signature__ = inspect.signature(res.__init__)\n",
-    "        return res\n",
-    "\n",
-    "    def __call__(cls, *args, **kwargs):\n",
-    "        f = first(args)\n",
-    "        n = getattr(f, '__name__', None)\n",
-    "        if _is_tfm_method(n, f):\n",
-    "            getattr(cls,n).add(f)\n",
-    "            return f\n",
-    "        obj = super().__call__(*args, **kwargs)\n",
-    "        # _TfmMeta.__new__ replaces cls.__signature__ which breaks the signature of a callable\n",
-    "        # instances of cls, fix it\n",
-    "        if hasattr(obj, '__call__'): obj.__signature__ = inspect.signature(obj.__call__)\n",
-    "        return obj\n",
-    "\n",
-    "    @classmethod\n",
-    "    def __prepare__(cls, name, bases): return _TfmDict()"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": null,