gemini-code-assist[bot] commented on code in PR #18574:
URL: https://github.com/apache/tvm/pull/18574#discussion_r2605966676
##########
python/tvm/relax/frontend/torch/exported_program_translator.py:
##########
@@ -1371,6 +1371,7 @@ def create_convert_map(
"any.dim": self._any,
"any.dims": self._any,
"mean.dim": self._mean,
+ "mean.default": self._mean,
Review Comment:
There is trailing whitespace on this line. Please remove it to adhere to PEP
8 style guidelines.
```suggestion
"mean.default": self._mean,
```
##########
tests/python/relax/test_frontend_from_exported_program.py:
##########
@@ -4910,6 +4910,10 @@ def forward(self, input):
class MeanKeepDim(Module):
def forward(self, input: torch.Tensor):
return input.mean(-1, keepdim=True)
+
Review Comment:
This line contains only whitespace. Please remove it to improve code
cleanliness and adhere to PEP 8.
##########
tests/python/relax/test_frontend_from_exported_program.py:
##########
@@ -4934,10 +4938,23 @@ def main(
gv: R.Tuple(R.Tensor((256, 1), dtype="float32")) = (lv,)
R.output(gv)
return gv
+
Review Comment:
This line contains only whitespace. Please remove it to improve code
cleanliness and adhere to PEP 8.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
