From 834e4e8a4473d5102f6442274ea945dccfcf8b05 Mon Sep 17 00:00:00 2001
From: Jeffrey Hu <hujh14@gmail.com>
Date: Mon, 28 Oct 2024 17:03:46 -0700
Subject: [PATCH] cleanup

---
 examples/{blur_kernel.py => blur_opt.py} | 0
 examples/simple_trainer.py               | 4 ++--
 2 files changed, 2 insertions(+), 2 deletions(-)
 rename examples/{blur_kernel.py => blur_opt.py} (100%)

diff --git a/examples/blur_kernel.py b/examples/blur_opt.py
similarity index 100%
rename from examples/blur_kernel.py
rename to examples/blur_opt.py
diff --git a/examples/simple_trainer.py b/examples/simple_trainer.py
index 8e369679..2e1eaa77 100644
--- a/examples/simple_trainer.py
+++ b/examples/simple_trainer.py
@@ -28,6 +28,7 @@
 from fused_ssim import fused_ssim
 from torchmetrics.image.lpip import LearnedPerceptualImagePatchSimilarity
 from typing_extensions import Literal, assert_never
+from blur_opt import BlurOptModule
 from utils import AppearanceOptModule, CameraOptModule, knn, rgb_to_sh, set_random_seed
 from lib_bilagrid import (
     BilateralGrid,
@@ -35,7 +36,6 @@
     color_correct,
     total_variation_loss,
 )
-from blur_kernel import BlurOptModule
 
 from gsplat.compression import PngCompression
 from gsplat.distributed import cli
@@ -412,7 +412,7 @@ def __init__(
 
         self.blur_optimizers = []
         if cfg.blur_opt:
-            self.blur_module = BlurOptModule(cfg, len(self.trainset)).to(self.device)
+            self.blur_module = BlurOptModule(len(self.trainset)).to(self.device)
             self.blur_module.zero_init()
             self.blur_optimizers = [
                 torch.optim.Adam(