Skip to content

Commit b81efbb

Browse files
committed
chore: minor type fixes
1 parent af54d23 commit b81efbb

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pygitguardian/client.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from .models import (
2020
Detail,
2121
Document,
22+
DocumentSchema,
2223
HealthCheckResponse,
2324
HoneytokenResponse,
2425
MultiScanResult,
@@ -252,7 +253,7 @@ def get(
252253
def post(
253254
self,
254255
endpoint: str,
255-
data: Optional[Dict[str, Any]] = None,
256+
data: Union[Dict[str, Any], List[Dict[str, Any]], None] = None,
256257
version: str = DEFAULT_API_VERSION,
257258
extra_headers: Optional[Dict[str, str]] = None,
258259
**kwargs: Any,
@@ -307,7 +308,7 @@ def content_scan(
307308
doc_dict["filename"] = filename
308309

309310
request_obj = Document.SCHEMA.load(doc_dict)
310-
Document.SCHEMA.validate_size(
311+
DocumentSchema.validate_size(
311312
request_obj, self.secret_scan_preferences.maximum_document_size
312313
)
313314

@@ -358,7 +359,7 @@ def multi_content_scan(
358359
raise TypeError("each document must be a dict")
359360

360361
for document in request_obj:
361-
Document.SCHEMA.validate_size(
362+
DocumentSchema.validate_size(
362363
document, self.secret_scan_preferences.maximum_document_size
363364
)
364365

0 commit comments

Comments
 (0)