Simply import pyrasite from our main module

42dbec38 · Luke Macken · a2d893d8 · 42dbec38
Commit 42dbec38 authored Mar 18, 2012 by Luke Macken
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

pyrasite/main.py pyrasite/main.py +3 -3

No files found.
--- a/pyrasite/main.py
+++ b/pyrasite/main.py
@@ -19,7 +19,7 @@ import os
 import sys
 import argparse
-from inject import CodeInjector
+import pyrasite
 def main():
@@ -58,8 +58,8 @@ def main():
        print("Error: The second argument must be a filename")
        sys.exit(4)
-    injector = CodeInjector(pid, verbose=args.verbose,
+    injector = pyrasite.CodeInjector(pid, verbose=args.verbose,
-                            gdb_prefix=args.gdb_prefix)
+                                     gdb_prefix=args.gdb_prefix)
    injector.inject(filename)