Skip to content

Commit 6876a0f

Browse files
committed
Fixed naming 🔤
1 parent ab8e47d commit 6876a0f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

scripts/make_scene_samples.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def get_z_indices_crop_out(z_indices: Tensor, predict_x: int, predict_y: int) ->
5252

5353
@torch.no_grad()
5454
def sample(model: Net2NetTransformer, annotations: List[Annotation], dataset: AnnotatedObjectsDataset,
55-
conditional_builder: CoordinatesCenterPointsConditionalBuilder, no_samples: int,
55+
conditional_builder: ObjectsCenterPointsConditionalBuilder, no_samples: int,
5656
temperature: float, top_k: int) -> Tensor:
5757
x_max, y_max = desired_z_shape[1], desired_z_shape[0]
5858

0 commit comments

Comments
 (0)