Import omni.usd ModuleNotFoundError: No module named 'omni.usd'

Hi, I am new to omniverse. I just want to write Python code to edit USD files. But I don’t know how to run this kind of code. I build a virtual environment to run the Python code. when I run the code in the virtual environment, there is an error as follows. Do I need to install some packages related to omni?
Python code:
from pxr import Usd, UsdGeom
import json
from functions import traverse_json
import omni.usd

import omni.usd
ModuleNotFoundError: No module named ‘omni.usd’

Thank you very much for your help.

I am not sure that is the right code to do this. Import is used to import a module not a usd file, so the error you are getting is as such. Let me get you the right code.