Open
Description
Describe the bug
Whenever sagemaker is imported, it seems to modify and automatically enable rich tracebacks.
To reproduce
Make a new environment
$ python -m venv .venv
$ source /.venv/bin/activate
$ python -m pip install sagemaker richMake a file that returns a traceback
"""rich_sage.py"""
import sagemaker
def main() -> None:
raise ValueError
if __name__ == "__main__":
main()Run the file
$ python rich_sage.py
sagemaker.config INFO - Not applying SDK defaults from location: /Library/Application Support/sagemaker/config.yaml
sagemaker.config INFO - Not applying SDK defaults from location: /Users/pawlu/Library/Application Support/sagemaker/config.yaml
╭─────────────────────────────── Traceback (most recent call last) ────────────────────────────────╮
│ /Users/pawlu/Documents/scratch/rich_sage/sage_me.py:9 in <module> │
│ │
│ 6 │
│ 7 │
│ 8 if __name__ == "__main__": │
│ ❱ 9 │ main() │
│ 10 │
│ │
│ /Users/pawlu/Documents/scratch/rich_sage/sage_me.py:5 in main │
│ │
│ 2 │
│ 3 │
│ 4 def main() -> None: │
│ ❱ 5 │ raise ValueError │
│ 6 │
│ 7 │
│ 8 if __name__ == "__main__": │
╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
ValueErrorIf the import statement is removed,
- import sagemakerthen the traceback renders as expected
$ python rich_sage.py
Traceback (most recent call last):
File "/Users/pawlu/Documents/scratch/rich_sage/sage_me.py", line 8, in <module>
main()
File "/Users/pawlu/Documents/scratch/rich_sage/sage_me.py", line 4, in main
raise ValueError
ValueErrorExpected behavior
Ideally, tracebacks shouldn't be modified without users input. At worst there should be a way to opt out of this. But we shouldn't be modifying tracebacks automatically at import. It's a tricky behavior that's hard to debug.
Screenshots or logs
System information
A description of your system. Please provide:
- SageMaker Python SDK version: 2.244.0
- Framework name (eg. PyTorch) or algorithm (eg. KMeans): Rich
- Framework version: 14.0.0
- Python version: 3.12.9
- CPU or GPU: M3 (macbook)
- Custom Docker image (Y/N): N
Thank you very much!