def initialize( self, get_examples: Callable[[], Iterable[Example]], *, nlp: Language = None ) -> None: lookups = Lookups() self._lookups = lookups.from_disk(path=self.source)
def test_lookups_to_from_disk(): lookups = Lookups() lookups.add_table("table1", {"foo": "bar", "hello": "world"}) lookups.add_table("table2", {"a": 1, "b": 2, "c": 3}) with make_tempdir() as tmpdir: lookups.to_disk(tmpdir) new_lookups = Lookups() new_lookups.from_disk(tmpdir) assert len(new_lookups) == 2 assert "table1" in new_lookups assert "table2" in new_lookups table1 = new_lookups.get_table("table1") assert len(table1) == 2 assert table1["foo"] == "bar" table2 = new_lookups.get_table("table2") assert len(table2) == 3 assert table2["b"] == 2
def from_disk(self, path, exclude=tuple()) -> "LookupLemmatizer": path: Path = ensure_path(path) lookups = Lookups() self._lookups = lookups.from_disk(path=path) return self