Skip to content

Commit 0e4d845

Browse files
committed
fix: check that a date has a doc before using note_datetime
1 parent 0c11230 commit 0e4d845

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

changelog.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
### Fixed
1010

1111
- Make catalogue & entrypoints compatible with py37-py312
12+
- Check that a data has a doc before trying to use the document's `note_datetime`
1213

1314
## v0.12.0
1415

edsnlp/pipes/misc/dates/models.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ def to_datetime(
125125
d.pop("mode", None)
126126
d.pop("bound", None)
127127

128-
if note_datetime is None:
128+
if note_datetime is None and self.doc is not None:
129129
note_datetime = self.doc._.note_datetime
130130

131131
if self.year and self.month and self.day:
@@ -178,7 +178,7 @@ def to_duration(
178178
note_datetime: Optional[Union[datetime.datetime, datetime.datetime]] = None,
179179
**kwargs,
180180
) -> Optional[datetime.timedelta]:
181-
if note_datetime is None:
181+
if note_datetime is None and self.doc is not None:
182182
note_datetime = self.doc._.note_datetime
183183

184184
if note_datetime and not isinstance(note_datetime, NaTType):
@@ -282,7 +282,7 @@ def to_datetime(
282282
note_datetime: Optional[Union[datetime.datetime, datetime.datetime]] = None,
283283
**kwargs,
284284
) -> Optional[datetime.datetime]:
285-
if note_datetime is None:
285+
if note_datetime is None and self.doc is not None:
286286
note_datetime = self.doc._.note_datetime
287287

288288
if note_datetime is not None and not isinstance(note_datetime, NaTType):

0 commit comments

Comments
 (0)