OLED not working with Python

[Vidit Katlana]

Hi,

I have Omega 2+ with Power dock and OLED Expansion.
I am trying to use OLED Display. It works perfect with console. But i am not able use it with Python coding.
I followed this Documentation - https://docs.onion.io/omega2-docs/oled-expansion-python-module.html#oled-py-set-memory-mode
And also tried to run this example - https://github.com/OnionIoT/i2c-exp-driver/blob/master/examples/oled-exp.py
The code did not work.
I am have full Python Installed I2C Module for Python installed and OLED Module for Python also installed.
Can someone help me in what I am missing out here ?