Skip to content

Commit dd00eca

Browse files
authored
Merge pull request #29 from IBM/python-change
update for python class name
2 parents 681bdda + 3597481 commit dd00eca

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

cldk/analysis/python/treesitter/python_sitter.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -197,6 +197,7 @@ def get_all_classes(self, module: str) -> List[PyClass]:
197197
full_signature=class_full_signature,
198198
methods=methods,
199199
super_classes=super_classes,
200+
class_name=klass_name,
200201
is_test_class=is_test_class))
201202
return classes
202203

cldk/models/python/models.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ class PyClass(BaseModel):
4545
full_signature: str
4646
super_classes: List[str]
4747
is_test_class: bool
48+
class_name: str = None
4849
methods: List[PyMethod]
4950

5051

0 commit comments

Comments
 (0)