small import bug

IPython/Extensions/numeric_formats.py

@@ -1,41 +1,43 b''
 # -*- coding: utf-8 -*-
 """
 Extension for printing Numeric Arrays in flexible ways.
 """
 
+from Numeric import ArrayType
+
 def num_display(self,arg):
     """Display method for printing which treats Numeric arrays specially.
     """
 
     # Non-numpy variables are printed using the system default
     if type(arg) != ArrayType:
         self._display(arg)
         return
     # Otherwise, we do work.
     format = __IPYTHON__.runtime_rc.numarray_print_format
     print 'NumPy array, format:',format
     # Here is where all the printing logic needs to be implemented
     print arg # nothing yet :)
 
 
 def magic_format(self,parameter_s=''):
     """Specifies format of numerical output.
 
     This command is similar to Ocave's format command.
     """
 
     valid_formats = ['long','short']
     
     if parameter_s in valid_formats:
         self.runtime_rc.numarray_print_format = parameter_s
         print 'Numeric output format is now:',parameter_s
     else:
         print 'Invalid format:',parameter_s
         print 'Valid formats:',valid_formats
 
 # setup default format
 __IPYTHON__.runtime_rc.numarray_print_format = 'long'
 
 # Bind our new functions to the interpreter
 __IPYTHON__.__class__.magic_format = magic_format
 __IPYTHON__.hooks.display = num_display