Fixed unit tests.

lu-ohai · lu-ohai · commit 30b8bcf14f96 · 2024-10-24T17:29:47.000-04:00
diff --git a/ads/aqua/modeldeployment/entities.py b/ads/aqua/modeldeployment/entities.py
@@ -3,7 +3,7 @@
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 from dataclasses import dataclass, field
-from typing import Union
+from typing import Union, Optional
 
 from oci.data_science.models import (
     ModelDeployment,
