Small bugfix in inheritance_diagram.py

mdboom · mdboom · commit 02680ee63b1b · 2008-08-12T18:36:29.000Z
svn path=/trunk/matplotlib/; revision=6028
diff --git a/doc/sphinxext/inheritance_diagram.py b/doc/sphinxext/inheritance_diagram.py
@@ -81,10 +81,9 @@ def _import_class_or_module(self, name):
             raise ValueError(
                 "Invalid class or module '%s' specified for inheritance diagram" % name)
         fullname = (path or '') + base
-        path = path and path.rstrip('.')
+        path = (path and path.rstrip('.'))
         if not path:
-            raise ValueError(
-                "Invalid class or module '%s' specified for inheritance diagram" % name)
+            path = base
         try:
             module = __import__(path, None, None, [])
         except ImportError:
