From c8b66980b2583423a97dd1e3840a63656d0d5652 Mon Sep 17 00:00:00 2001
From: Brenden Blanco <bblanco@plumgrid.com>
Date: Fri, 28 Aug 2015 23:15:19 -0700
Subject: [PATCH] Minor change to task_switch example syntax

Signed-off-by: Brenden Blanco <bblanco@plumgrid.com>
---
 README.md               | 6 ++----
 examples/task_switch.py | 3 +--
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index 29c74d0e..63451344 100644
--- a/README.md
+++ b/README.md
@@ -162,14 +162,12 @@ from bpf import BPF
 from time import sleep
 
 b = BPF(src_file="task_switch.c")
-fn = b.load_func("count_sched", BPF.KPROBE)
-stats = b.get_table("stats")
-BPF.attach_kprobe(fn, "finish_task_switch")
+b.attach_kprobe(event="finish_task_switch", fn_name="count_sched")
 
 # generate many schedule events
 for i in range(0, 100): sleep(0.01)
 
-for k, v in stats.items():
+for k, v in b["stats"].items():
     print("task_switch[%5d->%5d]=%u" % (k.prev_pid, k.curr_pid, v.value))
 ```
 [Source code listing](examples/task_switch.py)
diff --git a/examples/task_switch.py b/examples/task_switch.py
index 9eb4811c..c42d6377 100755
--- a/examples/task_switch.py
+++ b/examples/task_switch.py
@@ -6,11 +6,10 @@ from bpf import BPF
 from time import sleep
 
 b = BPF(src_file="task_switch.c")
-stats = b.get_table("stats")
 b.attach_kprobe(event="finish_task_switch", fn_name="count_sched")
 
 # generate many schedule events
 for i in range(0, 100): sleep(0.01)
 
-for k, v in stats.items():
+for k, v in b["stats"].items():
     print("task_switch[%5d->%5d]=%u" % (k.prev_pid, k.curr_pid, v.value))
-- 
GitLab