From c4431fe296fcf45addd5aefc082ac42b6e96a23a Mon Sep 17 00:00:00 2001
From: Matthew Tretter <m@tthewwithanm.com>
Date: Fri, 5 Apr 2013 16:41:53 -0400
Subject: [PATCH] Don't require generator to unregister

---
 imagekit/registry.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/imagekit/registry.py b/imagekit/registry.py
index 620f9f8..94adfdf 100644
--- a/imagekit/registry.py
+++ b/imagekit/registry.py
@@ -21,8 +21,7 @@ class GeneratorRegistry(object):
                                     ' already registered' % id)
         self._generators[id] = generator
 
-    def unregister(self, id, generator):
-        # TODO: Either don't require the generator, or--if we do--assert that it's registered with the provided id
+    def unregister(self, id):
         try:
             del self._generators[id]
         except KeyError:
@@ -177,8 +176,8 @@ class Unregister(object):
     Unregister generators and generated files.
 
     """
-    def generator(self, id, generator):
-        generator_registry.unregister(id, generator)
+    def generator(self, id):
+        generator_registry.unregister(id)
 
     def cachefiles(self, generator_id, cachefiles):
         cachefile_registry.unregister(generator_id, cachefiles)