Skip to content

Commit a2502e2

Browse files
committed
Revert old embedder[because of re-write]
1 parent bb7cc10 commit a2502e2

File tree

1 file changed

+13
-6
lines changed

1 file changed

+13
-6
lines changed

tutorials/embedder.ipynb

Lines changed: 13 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,7 @@
148148
"source": [
149149
"from adalflow.core.embedder import Embedder\n",
150150
"from adalflow.components.model_client import TransformersClient\n",
151+
"\n",
151152
"# from adalflow.utils import enable_library_logging\n",
152153
"\n",
153154
"# enable_library_logging(level=\"DEBUG\")\n",
@@ -499,32 +500,38 @@
499500
},
500501
{
501502
"cell_type": "code",
502-
"execution_count": null,
503+
"execution_count": 29,
503504
"metadata": {},
504505
"outputs": [],
505506
"source": [
506-
"from adalflow.core.types import Embedding, EmbedderOutput\n",
507+
"from adalflow.core.types import Embedding\n",
507508
"from adalflow.core.functional import normalize_vector\n",
508509
"from typing import List\n",
509510
"from adalflow.core.component import Component\n",
510511
"from copy import deepcopy\n",
512+
"\n",
513+
"\n",
511514
"class DecreaseEmbeddingDim(Component):\n",
512-
" def __init__(self, old_dim: int, new_dim: int, normalize: bool = True):\n",
515+
" def __init__(self, old_dim: int, new_dim: int, normalize: bool = True):\n",
513516
" super().__init__()\n",
514517
" self.old_dim = old_dim\n",
515518
" self.new_dim = new_dim\n",
516519
" self.normalize = normalize\n",
517520
" assert self.new_dim < self.old_dim, \"new_dim should be less than old_dim\"\n",
518521
"\n",
519522
" def call(self, input: List[Embedding]) -> List[Embedding]:\n",
520-
" output: EmbedderOutput = deepcopy(input)\n",
521-
" for embedding in output.data:\n",
523+
" output: List[Embedding] = deepcopy(input)\n",
524+
" for embedding in output:\n",
522525
" old_embedding = embedding.embedding\n",
523526
" new_embedding = old_embedding[: self.new_dim]\n",
524527
" if self.normalize:\n",
525528
" new_embedding = normalize_vector(new_embedding)\n",
526529
" embedding.embedding = new_embedding\n",
527-
" return output.data"
530+
" return output\n",
531+
"\n",
532+
" def _extra_repr(self) -> str:\n",
533+
" repr_str = f\"old_dim={self.old_dim}, new_dim={self.new_dim}, normalize={self.normalize}\"\n",
534+
" return repr_str"
528535
]
529536
},
530537
{

0 commit comments

Comments
 (0)