yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-08-29 19:06:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Unfreeze Document in Haystack 2.0 (#5974)

Browse Source 
* Unfreeze document

* Remove immutability test
This commit is contained in:
Vladimir Blagojevic 2023-10-05 17:55:07 +02:00 committed by GitHub
parent f983e605c7
commit 282419d82b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
haystack/preview/dataclasses/document.py
View File

@ -1,15 +1,13 @@
from typing import List, Any, Dict, Optional, Type
import json
import hashlib
import json
import logging
from dataclasses import asdict, dataclass, field, fields
from pathlib import Path
from dataclasses import dataclass, field, fields, asdict
from typing import Any, Dict, List, Optional, Type
import numpy
import pandas
logger = logging.getLogger(__name__)
@ -50,7 +48,7 @@ class DocumentDecoder(json.JSONDecoder):
return dictionary
@dataclass(frozen=True)
@dataclass
class Document:
"""
Base data class containing some data to be queried.

17
test/preview/dataclasses/test_document.py
View File

@ -1,21 +1,14 @@
from pathlib import Path
import dataclasses
import textwrap
import json
import textwrap
from pathlib import Path
import pytest
import pandas as pd
import numpy as np
import pandas as pd
import pytest
from haystack.preview import Document
from haystack.preview.dataclasses.document import DocumentEncoder, DocumentDecoder
@pytest.mark.unit
def test_document_is_immutable():
doc = Document(text="test text")
with pytest.raises(dataclasses.FrozenInstanceError):
doc.text = "won't work"
from haystack.preview.dataclasses.document import DocumentDecoder, DocumentEncoder
@pytest.mark.unit