From 83c018bb929244c2a00cf0cfac87984311bcda74 Mon Sep 17 00:00:00 2001
From: Casey-Shi <48111042+Yanxing-Shi@users.noreply.github.com>
Date: Wed, 28 May 2025 23:43:58 +0800
Subject: [PATCH] fix type hint (#2254)

[ROCm/composable_kernel commit: 4286eae09a148e9e86dc05a5c266123ec6cd931d]
---
 tile_engine/ops/gemm/json_config.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/tile_engine/ops/gemm/json_config.py b/tile_engine/ops/gemm/json_config.py
index f6303ec9f8..597caba76f 100644
--- a/tile_engine/ops/gemm/json_config.py
+++ b/tile_engine/ops/gemm/json_config.py
@@ -9,7 +9,7 @@ Handles loading, parsing, and validation of JSON configuration parameters.
 
 from pathlib import Path
 from dataclasses import dataclass
-from typing import List, Optional, Union, Tuple, Type
+from typing import List, Optional, Union, Tuple, Type, Dict
 import json
 
 
@@ -63,8 +63,8 @@ class ProblemConfig:
     layouts: Tuple[EnumConfigParam, ...]
 
     @property
-    def datatype_map(self) -> dict[str, str]:
-        """Get current layout selections as a key-value map."""
+    def datatype_map(self) -> Dict[str, str]:
+        """Get datatype as a key-value map."""
         return {
             'matrix_a': self.datatypes[0].values[0],
             'matrix_b': self.datatypes[1].values[0],
@@ -72,8 +72,8 @@ class ProblemConfig:
         }
 
     @property
-    def layout_map(self) -> dict[str, str]:
-        """Get current layout selections as a key-value map."""
+    def layout_map(self) -> Dict[str, str]:
+        """Get layout as a key-value map."""
         return {
             'matrix_a': self.layouts[0].values[0],
             'matrix_b': self.layouts[1].values[0],
@@ -83,7 +83,7 @@ class ProblemConfig:
 
 @dataclass
 class TileConfig:
-    """configuration class for tile parameter."""
+    """Configuration class for tile parameter."""
     tile_m: Union[EnumConfigParam, RangeConfigParam]
     tile_n: Union[EnumConfigParam, RangeConfigParam]
     tile_k: Union[EnumConfigParam, RangeConfigParam]
@@ -99,7 +99,7 @@ class TileConfig:
 
 @dataclass
 class TraitConfig:
-    """configuration class for kernel traits."""
+    """Configuration class for kernel traits."""
     pipeline: EnumConfigParam
     scheduler: EnumConfigParam
     epilogue: EnumConfigParam