format specialized_segment_ops_test.py to prepare D13515970 (#15408)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15408
Applied formatting to specialized_segment_ops_test.py to prepare D13515970
Reviewed By: salexspb
Differential Revision: D13520300
fbshipit-source-id: c3250b6abe8087c607f65ae60d1da61bd46c342b