Hi yes there is no problem in enrolling in driving school teaching lessons while you are pregnant. My wife took her lessons when she was in 6th month and tests untill she nearly finish the 8th month.
another thing you can register for automatic after taking permission from main traffic department in khalifa and you can redo your test every 15 days in case you fail your test. that was in Alrayah driving school
best of luck