Model editing for distribution shifts in uranium oxide morphological analysis