fix: add WindowsSessionUser dataclass to package exports

Signed-off-by: Yutong Li <[email protected]>
aws-deadline · Oct 28, 2024 · 5b6f6c8 · 5b6f6c8
1 parent 84592ad
commit 5b6f6c8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/deadline_test_fixtures/__init__.py b/src/deadline_test_fixtures/__init__.py
@@ -42,6 +42,7 @@
     JobAttachmentSettings,
     JobRunAsUser,
     PosixSessionUser,
+    WindowsSessionUser,
     S3Object,
     ServiceModel,
     OperatingSystem,
@@ -81,6 +82,7 @@
     "TaskStatus",
     "WindowsInstanceBuildWorker",
     "WindowsInstanceWorkerBase",
+    "WindowsSessionUser",
     "bootstrap_resources",
     "codeartifact",
     "deadline_client",

diff --git a/src/deadline_test_fixtures/models.py b/src/deadline_test_fixtures/models.py
@@ -39,8 +39,8 @@ class WindowsSessionUser:
 
 @dataclass(frozen=True)
 class JobRunAsUser:
-    posix: PosixSessionUser
-    windows: WindowsSessionUser
+    posix: PosixSessionUser | None
+    windows: WindowsSessionUser | None
     runAs: Literal["QUEUE_CONFIGURED_USER", "WORKER_AGENT_USER"]