Skip to content

Using only Kohlshuetter feature produces TypeError #94

@mtlive

Description

@mtlive

When I use kohlshuetter feature alone, trying to extract a content produces this error:

Traceback (most recent call last):
 File "test.py", line 35, in <module>
   content = content_extractor.extract(r.content)
 File "dragnet\extractor.py", line 171, in extract
   return str_cast(b'\n'.join(blocks[ind].text for ind in np.flatnonzero(preds)))
TypeError: sequence item 0: expected a bytes-like object, str found

I changed
str_cast(b'\n'.join(blocks[ind].text for ind in np.flatnonzero(preds)))
to
str_cast('\n'.join(blocks[ind].text for ind in np.flatnonzero(preds)))
and the error disappeared.
Is it ok to do so?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions