Restrict patching to correct Run class

[brimoor]

The patch_XXX() utils needed to restrict to the appropriate Run class.

from bson import ObjectId, DBRef
import numpy as np

import fiftyone as fo
import fiftyone.brain as fob
import fiftyone.zoo as foz
import fiftyone.core.odm as foo

db = foo.get_db_conn()

dataset = foz.load_zoo_dataset("quickstart", max_samples=10)

dataset.evaluate_detections(
    "predictions",
    gt_field="ground_truth",
    eval_key="eval",
)

fob.compute_similarity(
    dataset,
    embeddings=np.random.randn(len(dataset), 12),
    brain_key="sim",
)

dataset.save_view("test", dataset.limit(10))

# Break some references
dataset_dict = db.datasets.find_one({"name": dataset.name})
dataset_dict["saved_views"] = [ObjectId()]
dataset_dict["evaluations"] = {"eval": ObjectId()}
dataset_dict["brain_methods"] = {"sim": ObjectId()}
db.datasets.replace_one({"name": dataset.name}, dataset_dict)

# Bad IDs present as DBRefs
dataset.reload()
assert isinstance(dataset._doc.saved_views[0], DBRef)
assert isinstance(dataset._doc.evaluations["eval"], DBRef)
assert isinstance(dataset._doc.brain_methods["sim"], DBRef)

foo.patch_saved_views(dataset.name)
# Purging 1 bad saved view ID(s) {ObjectId('645d8f018fdea805437ee4b3')} from dataset
# Adding 1 misplaced saved view(s) [(ObjectId('645d8f018fdea805437ee4b2'), 'test')] back to dataset

foo.patch_evaluations(dataset.name)
# Purging 1 evaluation(s) {('eval', ObjectId('645d8f018fdea805437ee4b4'))} from dataset
# Adding 1 misplaced evaluation(s) {('eval', ObjectId('645d8f018fdea805437ee4ac'))} to dataset

foo.patch_brain_runs(dataset.name)
# Purging 1 brain method run(s) {('sim', ObjectId('645d8f018fdea805437ee4b5'))} from dataset
# Adding 1 misplaced brain method run(s) {('sim', ObjectId('645d8f018fdea805437ee4af'))} to dataset

dataset.reload()
assert len(dataset.list_saved_views()) == 1
assert len(dataset.list_evaluations()) == 1
assert len(dataset.list_brain_runs()) == 1