Adding return for the observer in the functional_modules.py
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/25168
Test Plan: Imported from OSS
Differential Revision: D17048164
Pulled By: zafartahirov
fbshipit-source-id: 40ee1f276ee5421255de5b2fc14194402ded10db