Correct problems with failing unit tests

Author: ErnestoLoma

pyatlan/client/entity.py

@@ -226,5 +226,5 @@ def get_entity_by_guid(
     def update_entity(self, entity: T) -> AssetMutationResponse[T]:
         request = BulkRequest[T](entities=[entity])
         raw_json = self.client.call_api(EntityClient.BULK_UPDATE, None, request)
-        response = AssetMutationResponse(**raw_json)
+        response: AssetMutationResponse[T] = AssetMutationResponse(**raw_json)
         return response

pyatlan/generator/generate_from_typdefs.py

@@ -16,8 +16,9 @@
 TEMPLATES_DIR = PARENT.parent / "templates"
 TYPE_DEF_FILE = Path(os.getenv("TMPDIR", "/tmp")) / "typedefs.json"
 TYPE_REPLACEMENTS = [
+    ("icon_type", "IconType"),
     ("string", "str"),
-    ("date", "date"),
+    ("date", "datetime"),
     ("array", "list"),
     ("boolean", "bool"),
     ("float", "float"),

pyatlan/generator/templates/entity.jinja2

@@ -1,10 +1,10 @@
 from __future__ import annotations
 from typing import Optional, Dict, Any, List, Literal
 from pydantic import Field
-from datetime import date
+from datetime import datetime
 from pyatlan.model.core import AtlanObject, Classification, Announcement
 from pyatlan.model.enums import CertificateStatus, EntityStatus, google_datastudio_asset_type, powerbi_endorsement, \
-    icon_type, AnnouncementType
+    IconType, AnnouncementType, SourceCostUnitType
 
 class Internal(AtlanObject):
     """For internal usage"""