pycln_library_remove_unused_imports_from_string.py

1import pycln
2from pathlib import Path
3
4# The source code with an unused import (e.g., 'os')
5source_code = """
6import os
7import sys
8
9print(sys.version)
10"""
11
12# Configure pycln settings
13# 'path' is required by the API even when processing strings
14config = pycln.Config(path=Path("."))
15
16# Remove unused imports
17cleaned_code = pycln.reformat_code(source_code, config)
18
19print(cleaned_code)
20# Output will have 'import os' removed